Home
last modified time | relevance | path

Searched refs:ShapeRepresentationFn (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.h74 const XlaHelpers::ShapeRepresentationFn shape_representation_fn = nullptr,
105 XlaHelpers::ShapeRepresentationFn shape_representation_fn,
115 XlaHelpers::ShapeRepresentationFn shape_representation_fn,
125 const XlaHelpers::ShapeRepresentationFn shape_representation_fn,
138 const XlaHelpers::ShapeRepresentationFn shape_representation_fn,
150 const XlaHelpers::ShapeRepresentationFn shape_representation_fn,
Dcompile_mlir_util.cc85 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in GetXlaInputShapes()
138 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in GetOutputInfo()
392 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ConvertMLIRToXlaComputation()
408 XlaHelpers::ShapeRepresentationFn* shape_representation_fn) { in CompileMlirSetup()
431 XlaHelpers::ShapeRepresentationFn shape_representation_fn; in BuildHloFromTf()
449 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in PopulateResultIOInfo()
471 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileMlirToXlaHlo()
493 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileSerializedMlirToXlaHlo()
634 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileGraphToXlaHlo()
701 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileGraphToXlaHlo()
Dtf_xla_mlir_translate.cc286 XlaHelpers::ShapeRepresentationFn shape_representation_fn = in CompileMlirToXlaHloViaBuilder()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h79 ShapeRepresentationFn; typedef
83 XlaHelpers::ShapeRepresentationFn IdentityShapeRepresentationFn();
88 XlaHelpers::ShapeRepresentationFn shape_representation_fn,
95 XlaHelpers::ShapeRepresentationFn shape_representation_fn,
Dxla_helpers.cc129 XlaHelpers::ShapeRepresentationFn IdentityShapeRepresentationFn() { in IdentityShapeRepresentationFn()
141 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in RewriteLayoutWithShardedShape()
182 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ReshapeWithCorrectRepresentationAndSharding()
Dxla_compiler.h143 ShapeRepresentationFn; typedef
176 ShapeRepresentationFn shape_representation_fn;
Dxla_compiler.cc171 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn, in BuildComputation()
1388 : ShapeRepresentationFn{}, in CompileGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h63 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
72 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn()
83 XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
128 XlaCompiler::ShapeRepresentationFn shape_representation_fn;
245 const XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
Dxla_device_context.h60 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
82 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn()
114 XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
Dxla_device_context.cc82 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in XlaDeviceContext()
Dxla_device.cc148 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in Metadata()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.h174 const XlaCompiler::ShapeRepresentationFn shape_representation_fn,
188 const XlaCompiler::ShapeRepresentationFn shape_representation_fn,
Dtpu_compile_op_common.cc257 const XlaCompiler::ShapeRepresentationFn shape_representation_fn, in GetShardingInfo()
280 const XlaCompiler::ShapeRepresentationFn shape_representation_fn, in CompileTFFunctionToHlo()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.h51 const tensorflow::XlaHelpers::ShapeRepresentationFn
Dmlir_hlo_to_hlo.cc446 tensorflow::XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ConvertToHloModule()
538 tensorflow::XlaHelpers::ShapeRepresentationFn shape_representation_fn_;
1769 const tensorflow::XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ConvertMlirHloToHlo()