Home
last modified time | relevance | path

Searched defs:shape_determination_fns (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc99 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in MaybeRewriteLayoutWithShardedShape()
118 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in GetXlaInputShapes()
187 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in GetOutputInfo()
520 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in ConvertMLIRToXlaComputation()
617 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in PopulateResultIOInfo()
639 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in CompileMlirToXlaHlo()
668 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in CompileSerializedMlirToXlaHlo()
823 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in CompileGraphToXlaHlo()
880 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in CompileGraphToXlaHlo()
Dtf_xla_mlir_translate.cc302 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns{ in CompileMlirToXlaHloViaBuilder() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dlayout_util.cc39 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in RewriteLayoutWithShardedShape()
84 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in ReshapeWithCorrectRepresentationAndSharding()
Dxla_compiler.h172 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; member
Dxla_compiler_test.cc307 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
352 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
408 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
1169 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
1212 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
1348 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
1421 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns; in TEST_F() local
Dxla_compiler.cc173 const XlaShapeLayoutHelpers::ShapeDeterminationFns& shape_determination_fns, in BuildComputation()
525 XlaShapeLayoutHelpers::ShapeDeterminationFns& shape_determination_fns = in XlaCompiler() local
1417 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns{ in CompileGraph() local
Dxla_op_kernel.cc569 auto shape_determination_fns = in ReadVariableInputTensor() local
717 auto shape_determination_fns = in AssignVariableTensor() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.h84 const XlaShapeLayoutHelpers::ShapeDeterminationFns& shape_determination_fns() in shape_determination_fns() function
Dxla_tpu_device.cc203 const auto& shape_determination_fns = in TpuDeviceToDeviceCopy() local
385 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns{ in CreateDevices() local
Dxla_device.h138 shape_determination_fns; member
Dxla_device_context.cc93 XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in XlaDeviceContext()
Dxla_device.cc147 std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns> in Metadata()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_compile.cc373 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in CompileTFFunctionToHlo()
454 const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, in GetShardingInfo()