Update Dataset API documentation.
PiperOrigin-RevId: 163349457
This commit is contained in:
parent
e4a5c53560
commit
34cbf161d7
tensorflow
compiler/xla
client
service
cpu
compiler_functor.cccompiler_functor.hcpu_compiler.cccpu_executable.ccdisassembler.ccdisassembler.hdot_op_emitter.ccdot_op_emitter.helemental_ir_emitter.ccelemental_ir_emitter.hir_emitter.ccir_emitter.hparallel_cpu_executable.ccsimple_orc_jit.ccsimple_orc_jit.h
elemental_ir_emitter.ccelemental_ir_emitter.hgpu
convolution_folding.ccelemental_ir_emitter.ccelemental_ir_emitter.hgpu_compiler.cchlo_to_ir_bindings.cchlo_to_ir_bindings.hir_emission_utils.ccir_emission_utils.hir_emitter.ccir_emitter.hir_emitter_context.hir_emitter_nested.ccir_emitter_unnested.cc
llvm_gpu_backend
parallel_loop_emitter.ccparallel_loop_emitter.hllvm_ir
tests
docs_src/programmers_guide
@ -15,7 +15,7 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/compiler/xla/client/compile_only_client.h"
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/Triple.h"
|
||||
#include "llvm/ADT/Triple.h"
|
||||
#include "tensorflow/compiler/xla/ptr_util.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/status_macros.h"
|
||||
|
@ -17,7 +17,7 @@ limitations under the License.
|
||||
|
||||
#include <utility>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/Triple.h"
|
||||
#include "llvm/ADT/Triple.h"
|
||||
#include "tensorflow/compiler/xla/ptr_util.h"
|
||||
#include "tensorflow/compiler/xla/service/backend.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
|
@ -22,19 +22,19 @@ limitations under the License.
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/StringRef.h"
|
||||
#include "external/llvm/include/llvm/Analysis/TargetLibraryInfo.h"
|
||||
#include "external/llvm/include/llvm/Analysis/TargetTransformInfo.h"
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/ObjectMemoryBuffer.h"
|
||||
#include "external/llvm/include/llvm/IR/LegacyPassManager.h"
|
||||
#include "external/llvm/include/llvm/IR/Verifier.h"
|
||||
#include "external/llvm/include/llvm/MC/MCContext.h"
|
||||
#include "external/llvm/include/llvm/Object/ObjectFile.h"
|
||||
#include "external/llvm/include/llvm/Support/raw_ostream.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetMachine.h"
|
||||
#include "external/llvm/include/llvm/Transforms/IPO.h"
|
||||
#include "external/llvm/include/llvm/Transforms/IPO/AlwaysInliner.h"
|
||||
#include "external/llvm/include/llvm/Transforms/IPO/PassManagerBuilder.h"
|
||||
#include "llvm/ADT/StringRef.h"
|
||||
#include "llvm/Analysis/TargetLibraryInfo.h"
|
||||
#include "llvm/Analysis/TargetTransformInfo.h"
|
||||
#include "llvm/ExecutionEngine/ObjectMemoryBuffer.h"
|
||||
#include "llvm/IR/LegacyPassManager.h"
|
||||
#include "llvm/IR/Verifier.h"
|
||||
#include "llvm/MC/MCContext.h"
|
||||
#include "llvm/Object/ObjectFile.h"
|
||||
#include "llvm/Support/raw_ostream.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "llvm/Transforms/IPO.h"
|
||||
#include "llvm/Transforms/IPO/AlwaysInliner.h"
|
||||
#include "llvm/Transforms/IPO/PassManagerBuilder.h"
|
||||
#include "tensorflow/compiler/xla/ptr_util.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_runtime.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_runtime_avx.h"
|
||||
|
@ -16,10 +16,10 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_COMPILER_FUNCTOR_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_COMPILER_FUNCTOR_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/LegacyPassManager.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/Object/ObjectFile.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetMachine.h"
|
||||
#include "llvm/IR/LegacyPassManager.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/Object/ObjectFile.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/disassembler.h"
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
|
||||
|
@ -26,17 +26,17 @@ limitations under the License.
|
||||
|
||||
// IWYU pragma: no_include "llvm/Config/Disassemblers.def.inc"
|
||||
// IWYU pragma: no_include "llvm/Config/Targets.def.inc"
|
||||
#include "external/llvm/include/llvm/ADT/StringRef.h"
|
||||
#include "external/llvm/include/llvm/ADT/Triple.h"
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/Object/ObjectFile.h"
|
||||
#include "external/llvm/include/llvm/Support/CommandLine.h"
|
||||
#include "external/llvm/include/llvm/Support/TargetRegistry.h"
|
||||
#include "external/llvm/include/llvm/Support/TargetSelect.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetMachine.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetOptions.h"
|
||||
#include "llvm/ADT/StringRef.h"
|
||||
#include "llvm/ADT/Triple.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/Object/ObjectFile.h"
|
||||
#include "llvm/Support/CommandLine.h"
|
||||
#include "llvm/Support/TargetRegistry.h"
|
||||
#include "llvm/Support/TargetSelect.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "llvm/Target/TargetOptions.h"
|
||||
#include "tensorflow/compiler/xla/literal_util.h"
|
||||
#include "tensorflow/compiler/xla/map_util.h"
|
||||
#include "tensorflow/compiler/xla/protobuf_util.h"
|
||||
|
@ -22,7 +22,7 @@ limitations under the License.
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
||||
#include "llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/computation_layout.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_computation.h"
|
||||
|
@ -21,9 +21,9 @@ limitations under the License.
|
||||
#include <type_traits>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/MC/MCInst.h"
|
||||
#include "external/llvm/include/llvm/Support/TargetRegistry.h"
|
||||
#include "external/llvm/include/llvm/Support/raw_ostream.h"
|
||||
#include "llvm/MC/MCInst.h"
|
||||
#include "llvm/Support/TargetRegistry.h"
|
||||
#include "llvm/Support/raw_ostream.h"
|
||||
#include "tensorflow/compiler/xla/status_macros.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/compiler/xla/util.h"
|
||||
|
@ -19,14 +19,14 @@ limitations under the License.
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
||||
#include "external/llvm/include/llvm/MC/MCContext.h"
|
||||
#include "external/llvm/include/llvm/MC/MCDisassembler/MCDisassembler.h"
|
||||
#include "external/llvm/include/llvm/MC/MCInstPrinter.h"
|
||||
#include "external/llvm/include/llvm/MC/MCInstrAnalysis.h"
|
||||
#include "external/llvm/include/llvm/MC/MCObjectFileInfo.h"
|
||||
#include "external/llvm/include/llvm/MC/MCSubtargetInfo.h"
|
||||
#include "external/llvm/include/llvm/Object/ObjectFile.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetMachine.h"
|
||||
#include "llvm/MC/MCContext.h"
|
||||
#include "llvm/MC/MCDisassembler/MCDisassembler.h"
|
||||
#include "llvm/MC/MCInstPrinter.h"
|
||||
#include "llvm/MC/MCInstrAnalysis.h"
|
||||
#include "llvm/MC/MCObjectFileInfo.h"
|
||||
#include "llvm/MC/MCSubtargetInfo.h"
|
||||
#include "llvm/Object/ObjectFile.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "tensorflow/compiler/xla/statusor.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
||||
|
@ -18,10 +18,10 @@ limitations under the License.
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_runtime.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/ir_emission_utils.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_module.h"
|
||||
|
@ -16,7 +16,7 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_DOT_OP_EMITTER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_DOT_OP_EMITTER_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_module_config.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
|
@ -17,8 +17,8 @@ limitations under the License.
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_opcode.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
@ -16,9 +16,9 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_ELEMENTAL_IR_EMITTER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_ELEMENTAL_IR_EMITTER_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/elemental_ir_emitter.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
#include "tensorflow/compiler/xla/statusor.h"
|
||||
|
@ -26,14 +26,14 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
// IWYU pragma: no_include "llvm/IR/Intrinsics.gen.inc"
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Constants.h"
|
||||
#include "external/llvm/include/llvm/IR/GlobalVariable.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Intrinsics.h"
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetRegisterInfo.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetSubtargetInfo.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Constants.h"
|
||||
#include "llvm/IR/GlobalVariable.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Intrinsics.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/Target/TargetRegisterInfo.h"
|
||||
#include "llvm/Target/TargetSubtargetInfo.h"
|
||||
#include "tensorflow/compiler/xla/layout_util.h"
|
||||
#include "tensorflow/compiler/xla/map_util.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
|
@ -22,11 +22,11 @@ limitations under the License.
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/Triple.h"
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/ADT/Triple.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/dfs_hlo_visitor_with_default.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_computation.h"
|
||||
|
@ -24,7 +24,7 @@ limitations under the License.
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
||||
#include "llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
||||
#include "tensorflow/compiler/xla/map_util.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_runtime.h"
|
||||
|
@ -21,11 +21,11 @@ limitations under the License.
|
||||
#include <list>
|
||||
#include <utility>
|
||||
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/ExecutionEngine.h"
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/SectionMemoryManager.h"
|
||||
#include "external/llvm/include/llvm/IR/Mangler.h"
|
||||
#include "external/llvm/include/llvm/Support/CodeGen.h"
|
||||
#include "external/llvm/include/llvm/Support/Host.h"
|
||||
#include "llvm/ExecutionEngine/ExecutionEngine.h"
|
||||
#include "llvm/ExecutionEngine/SectionMemoryManager.h"
|
||||
#include "llvm/IR/Mangler.h"
|
||||
#include "llvm/Support/CodeGen.h"
|
||||
#include "llvm/Support/Host.h"
|
||||
#include "tensorflow/compiler/xla/ptr_util.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_runtime.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_runtime_avx.h"
|
||||
|
@ -20,11 +20,11 @@ limitations under the License.
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/Triple.h"
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
||||
#include "external/llvm/include/llvm/ExecutionEngine/Orc/RTDyldObjectLinkingLayer.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetMachine.h"
|
||||
#include "llvm/ADT/Triple.h"
|
||||
#include "llvm/ExecutionEngine/Orc/IRCompileLayer.h"
|
||||
#include "llvm/ExecutionEngine/Orc/RTDyldObjectLinkingLayer.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/compiler_functor.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/disassembler.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
@ -21,10 +21,10 @@ limitations under the License.
|
||||
#include <vector>
|
||||
|
||||
// IWYU pragma: no_include "llvm/IR/Intrinsics.gen.inc"
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Intrinsics.h"
|
||||
#include "external/llvm/include/llvm/Transforms/Utils/BasicBlockUtils.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Intrinsics.h"
|
||||
#include "llvm/Transforms/Utils/BasicBlockUtils.h"
|
||||
#include "tensorflow/compiler/xla/primitive_util.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_module.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_opcode.h"
|
||||
|
@ -18,9 +18,9 @@ limitations under the License.
|
||||
|
||||
#include <unordered_map>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_module_config.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h"
|
||||
|
@ -43,7 +43,7 @@ MatchBackwardFilter(HloInstruction* conv) {
|
||||
ConvolutionDimensionNumbers());
|
||||
// Step 1: match the instruction pattern without considering the paddings and
|
||||
// dimension numbers just yet. We may need some generic pattern matcher
|
||||
// similar to external/llvm/include/llvm/IR/PatternMatch.h
|
||||
// similar to third_party/llvm/llvm/include/llvm/IR/PatternMatch.h
|
||||
//
|
||||
// Backward filter convolution is implemented in XLA as the forward
|
||||
// convolution of padded activations and dilated gradients. Padding on
|
||||
|
@ -23,12 +23,12 @@ limitations under the License.
|
||||
#include "tensorflow/core/platform/types.h"
|
||||
// IWYU pragma: no_include "llvm/IR/Attributes.gen.inc"
|
||||
// IWYU pragma: no_include "llvm/IR/Intrinsics.gen.inc"
|
||||
#include "external/llvm/include/llvm/ADT/APInt.h"
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Intrinsics.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IR/Type.h"
|
||||
#include "llvm/ADT/APInt.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Intrinsics.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/Type.h"
|
||||
#include "tensorflow/compiler/xla/literal_util.h"
|
||||
#include "tensorflow/compiler/xla/primitive_util.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_opcode.h"
|
||||
|
@ -20,8 +20,8 @@ limitations under the License.
|
||||
#include <string>
|
||||
#include <utility>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/elemental_ir_emitter.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_computation.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
|
@ -19,10 +19,10 @@ limitations under the License.
|
||||
#include <functional>
|
||||
#include <utility>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/DiagnosticInfo.h"
|
||||
#include "external/llvm/include/llvm/IR/DiagnosticPrinter.h"
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/DiagnosticInfo.h"
|
||||
#include "llvm/IR/DiagnosticPrinter.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/protobuf_util.h"
|
||||
#include "tensorflow/compiler/xla/ptr_util.h"
|
||||
#include "tensorflow/compiler/xla/service/algebraic_simplifier.h"
|
||||
|
@ -15,9 +15,9 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h"
|
||||
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/ir_emission_utils.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_opcode.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
|
@ -18,8 +18,8 @@ limitations under the License.
|
||||
|
||||
#include <unordered_map>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/map_util.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
|
@ -18,7 +18,7 @@ limitations under the License.
|
||||
#include <algorithm>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/layout_util.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_computation.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
|
@ -18,8 +18,8 @@ limitations under the License.
|
||||
|
||||
#include <utility>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
|
||||
namespace xla {
|
||||
|
@ -20,10 +20,10 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
// IWYU pragma: no_include "llvm/IR/Intrinsics.gen.inc"
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Constants.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Constants.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/primitive_util.h"
|
||||
#include "tensorflow/compiler/xla/service/elemental_ir_emitter.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/elemental_ir_emitter.h"
|
||||
|
@ -35,9 +35,9 @@ limitations under the License.
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/dfs_hlo_visitor_with_default.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/elemental_ir_emitter.h"
|
||||
|
@ -16,7 +16,7 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_GPU_IR_EMITTER_CONTEXT_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_GPU_IR_EMITTER_CONTEXT_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/partition_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/name_uniquer.h"
|
||||
|
@ -16,10 +16,10 @@ limitations under the License.
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/hlo_to_ir_bindings.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/ir_emitter.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/ir_emitter_context.h"
|
||||
|
@ -17,13 +17,13 @@ limitations under the License.
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/StringRef.h"
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/ADT/StringRef.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/literal_util.h"
|
||||
#include "tensorflow/compiler/xla/ptr_util.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
|
@ -15,9 +15,9 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/dump_ir_pass.h"
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/Support/FileSystem.h"
|
||||
#include "external/llvm/include/llvm/Support/raw_ostream.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/Support/FileSystem.h"
|
||||
#include "llvm/Support/raw_ostream.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/utils.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
|
@ -18,8 +18,8 @@ limitations under the License.
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/LegacyPassManager.h"
|
||||
#include "external/llvm/include/llvm/Pass.h"
|
||||
#include "llvm/IR/LegacyPassManager.h"
|
||||
#include "llvm/Pass.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
||||
namespace xla {
|
||||
|
@ -27,33 +27,33 @@ limitations under the License.
|
||||
#include "tensorflow/compiler/xla/status_macros.h"
|
||||
#include "tensorflow/compiler/xla/util.h"
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/STLExtras.h"
|
||||
#include "external/llvm/include/llvm/ADT/StringMap.h"
|
||||
#include "external/llvm/include/llvm/ADT/StringSet.h"
|
||||
#include "external/llvm/include/llvm/Analysis/TargetLibraryInfo.h"
|
||||
#include "external/llvm/include/llvm/Analysis/TargetTransformInfo.h"
|
||||
#include "external/llvm/include/llvm/Bitcode/BitcodeReader.h"
|
||||
#include "external/llvm/include/llvm/Bitcode/BitcodeWriter.h"
|
||||
#include "external/llvm/include/llvm/CodeGen/CommandFlags.h"
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/IR/LegacyPassManager.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/LinkAllIR.h"
|
||||
#include "external/llvm/include/llvm/LinkAllPasses.h"
|
||||
#include "external/llvm/include/llvm/Linker/Linker.h"
|
||||
#include "external/llvm/include/llvm/PassRegistry.h"
|
||||
#include "external/llvm/include/llvm/Support/CommandLine.h"
|
||||
#include "external/llvm/include/llvm/Support/FileSystem.h"
|
||||
#include "external/llvm/include/llvm/Support/FormattedStream.h"
|
||||
#include "external/llvm/include/llvm/Support/TargetRegistry.h"
|
||||
#include "external/llvm/include/llvm/Support/TargetSelect.h"
|
||||
#include "external/llvm/include/llvm/Support/ToolOutputFile.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetMachine.h"
|
||||
#include "external/llvm/include/llvm/Transforms/IPO.h"
|
||||
#include "external/llvm/include/llvm/Transforms/IPO/AlwaysInliner.h"
|
||||
#include "external/llvm/include/llvm/Transforms/IPO/PassManagerBuilder.h"
|
||||
#include "llvm/ADT/STLExtras.h"
|
||||
#include "llvm/ADT/StringMap.h"
|
||||
#include "llvm/ADT/StringSet.h"
|
||||
#include "llvm/Analysis/TargetLibraryInfo.h"
|
||||
#include "llvm/Analysis/TargetTransformInfo.h"
|
||||
#include "llvm/Bitcode/BitcodeReader.h"
|
||||
#include "llvm/Bitcode/BitcodeWriter.h"
|
||||
#include "llvm/CodeGen/CommandFlags.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/LegacyPassManager.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/LinkAllIR.h"
|
||||
#include "llvm/LinkAllPasses.h"
|
||||
#include "llvm/Linker/Linker.h"
|
||||
#include "llvm/PassRegistry.h"
|
||||
#include "llvm/Support/CommandLine.h"
|
||||
#include "llvm/Support/FileSystem.h"
|
||||
#include "llvm/Support/FormattedStream.h"
|
||||
#include "llvm/Support/TargetRegistry.h"
|
||||
#include "llvm/Support/TargetSelect.h"
|
||||
#include "llvm/Support/ToolOutputFile.h"
|
||||
#include "llvm/Target/TargetMachine.h"
|
||||
#include "llvm/Transforms/IPO.h"
|
||||
#include "llvm/Transforms/IPO/AlwaysInliner.h"
|
||||
#include "llvm/Transforms/IPO/PassManagerBuilder.h"
|
||||
|
||||
#include "external/llvm/include/llvm/Transforms/IPO/Internalize.h"
|
||||
#include "llvm/Transforms/IPO/Internalize.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/io/path.h"
|
||||
@ -178,7 +178,7 @@ std::unique_ptr<llvm::TargetMachine> GetTargetMachine(
|
||||
target_options.MCOptions.AsmVerbose = false;
|
||||
|
||||
// The selection of codegen optimization level is copied from function
|
||||
// GetCodeGenOptLevel in //external/llvm/tools/opt/opt.cpp.
|
||||
// GetCodeGenOptLevel in //third_party/llvm/llvm/tools/opt/opt.cpp.
|
||||
CodeGenOpt::Level codegen_opt_level;
|
||||
switch (hlo_module_config.debug_options().xla_backend_optimization_level()) {
|
||||
case 1:
|
||||
|
@ -20,7 +20,7 @@ limitations under the License.
|
||||
#include <string>
|
||||
#include <utility>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_module_config.h"
|
||||
#include "tensorflow/compiler/xla/statusor.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
@ -17,10 +17,10 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IRReader/IRReader.h"
|
||||
#include "external/llvm/include/llvm/Support/SourceMgr.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IRReader/IRReader.h"
|
||||
#include "llvm/Support/SourceMgr.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/core/lib/core/stringpiece.h"
|
||||
#include "tensorflow/core/lib/strings/strcat.h"
|
||||
|
@ -19,8 +19,8 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/core/lib/io/path.h"
|
||||
|
||||
#include "external/llvm/include/llvm/IR/LLVMContext.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/LLVMContext.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/core/platform/test.h"
|
||||
|
||||
|
@ -20,8 +20,8 @@ limitations under the License.
|
||||
#include "tensorflow/core/platform/logging.h"
|
||||
#include "tensorflow/core/platform/types.h"
|
||||
// IWYU pragma: no_include "llvm/IR/Intrinsics.gen.inc"
|
||||
#include "external/llvm/include/llvm/IR/Intrinsics.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/Intrinsics.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_loop.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/shape_util.h"
|
||||
|
@ -16,7 +16,7 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_GPU_PARALLEL_LOOP_EMITTER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_GPU_PARALLEL_LOOP_EMITTER_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "tensorflow/compiler/xla/service/gpu/partition_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/loop_emitter.h"
|
||||
|
@ -17,7 +17,7 @@ limitations under the License.
|
||||
|
||||
#include <unordered_set>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/MDBuilder.h"
|
||||
#include "llvm/IR/MDBuilder.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/service/logical_buffer.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
@ -16,7 +16,7 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_ALIAS_ANALYSIS_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_ALIAS_ANALYSIS_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "tensorflow/compiler/xla/service/buffer_assignment.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
|
@ -17,8 +17,8 @@ limitations under the License.
|
||||
|
||||
#include <functional>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/elemental_ir_emitter.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
|
@ -19,8 +19,8 @@ limitations under the License.
|
||||
#include <map>
|
||||
#include <unordered_map>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/dfs_hlo_visitor_with_default.h"
|
||||
#include "tensorflow/compiler/xla/service/elemental_ir_emitter.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
|
||||
|
@ -15,8 +15,8 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Constants.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Constants.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "tensorflow/compiler/xla/layout_util.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/shape_util.h"
|
||||
|
@ -19,8 +19,8 @@ limitations under the License.
|
||||
#include <map>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/map_util.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/compiler/xla/xla_data.pb.h"
|
||||
|
@ -18,9 +18,9 @@ limitations under the License.
|
||||
#include <numeric>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Constants.h"
|
||||
#include "external/llvm/include/llvm/IR/Function.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Constants.h"
|
||||
#include "llvm/IR/Function.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/shape_util.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
@ -19,9 +19,9 @@ limitations under the License.
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/compiler/xla/xla_data.pb.h"
|
||||
|
@ -18,9 +18,9 @@ limitations under the License.
|
||||
#include <algorithm>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/MDBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Operator.h"
|
||||
#include "external/llvm/include/llvm/Target/TargetOptions.h"
|
||||
#include "llvm/IR/MDBuilder.h"
|
||||
#include "llvm/IR/Operator.h"
|
||||
#include "llvm/Target/TargetOptions.h"
|
||||
#include "tensorflow/compiler/xla/layout_util.h"
|
||||
#include "tensorflow/compiler/xla/literal_util.h"
|
||||
#include "tensorflow/compiler/xla/shape_util.h"
|
||||
|
@ -20,13 +20,13 @@ limitations under the License.
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/StringRef.h"
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "external/llvm/include/llvm/IR/Module.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "external/llvm/include/llvm/Support/raw_ostream.h"
|
||||
#include "llvm/ADT/StringRef.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "llvm/Support/raw_ostream.h"
|
||||
#include "tensorflow/compiler/xla/literal_util.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
#include "tensorflow/compiler/xla/xla_data.pb.h"
|
||||
|
@ -18,9 +18,9 @@ limitations under the License.
|
||||
|
||||
#include <functional>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/BasicBlock.h"
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/BasicBlock.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
#include "tensorflow/compiler/xla/statusor.h"
|
||||
|
||||
|
@ -19,7 +19,7 @@ limitations under the License.
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/IR/Instructions.h"
|
||||
#include "llvm/IR/Instructions.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
|
||||
#include "tensorflow/compiler/xla/shape_util.h"
|
||||
#include "tensorflow/compiler/xla/types.h"
|
||||
|
@ -16,8 +16,8 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_OPS_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_OPS_H_
|
||||
|
||||
#include "external/llvm/include/llvm/IR/IRBuilder.h"
|
||||
#include "external/llvm/include/llvm/IR/Value.h"
|
||||
#include "llvm/IR/IRBuilder.h"
|
||||
#include "llvm/IR/Value.h"
|
||||
#include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
|
||||
#include "tensorflow/core/lib/gtl/array_slice.h"
|
||||
#include "tensorflow/core/platform/types.h"
|
||||
|
@ -19,7 +19,7 @@ limitations under the License.
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
|
||||
#include "external/llvm/include/llvm/ADT/Triple.h"
|
||||
#include "llvm/ADT/Triple.h"
|
||||
#include "tensorflow/compiler/xla/client/client_library.h"
|
||||
#include "tensorflow/compiler/xla/client/computation_builder.h"
|
||||
#include "tensorflow/compiler/xla/service/cpu/cpu_compiler.h"
|
||||
|
@ -122,7 +122,7 @@ dataset3 = dataset3.filter(lambda x, (y, z): ...)
|
||||
|
||||
One you have built a `Dataset` to represent your input data, the next step is to
|
||||
create an `Iterator` to access elements from that dataset. The `Dataset` API
|
||||
currently supports three kinds of iterator, in increasing level of
|
||||
currently supports the following iterators, in increasing level of
|
||||
sophistication:
|
||||
|
||||
* **one-shot**,
|
||||
|
Loading…
Reference in New Issue
Block a user