| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_layout.h | 36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout()
67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout() function
|
| D | compile_only_service.cc | 110 Shape result_layout(instance.result_layout); in CompileAheadOfTime() local
|
| D | compile_only_service.h | 45 Shape result_layout; member
|
| D | layout_assignment.cc | 805 const ShapeLayout* result_layout = constraints->ResultLayout(); in AddMandatoryConstraints() local
1228 const ShapeLayout& result_layout = in CheckLayouts() local
2398 ShapeLayout* result_layout = computation_layout->mutable_result_layout(); in PropagateComputationLayouts() local
|
| D | hlo_verifier.cc | 1812 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout() local
2515 const Layout& result_layout = result_shape.layout(); in Postprocess() local
|
| D | hlo_parser_test.cc | 2890 auto result_layout = program_layout.result_layout().layout(); local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 58 const Shape* ExecutableBuildOptions::result_layout() const { in result_layout() function in xla::ExecutableBuildOptions
169 std::string result_layout = "nullopt"; in ToString() local
|
| D | compile_only_client.h | 46 const Shape* result_layout; member
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | compile_options.proto | 23 xla.ShapeProto result_layout = 2; field
|
| D | utils.cc | 199 Shape result_layout; in DetermineArgumentLayoutsFromCompileOptions() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 73 LayoutProto result_layout = 2; field
|
| D | jitrt_custom_calls.h | 67 llvm::ArrayRef<int64_t> result_layout; member
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | cluster_function_conversion.cc | 71 auto result_layout = result_layout_or_status.ValueOrDie(); in AttachRetvalLayouts() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 60 absl::optional<const Shape> result_layout, in CreateModuleConfig()
139 absl::optional<const Shape> result_layout, in CreateModuleConfig()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_conversions.cc | 443 xla::ShapeLayout result_layout( in ConfigWithLayout() local
|
| D | c_api_decl.h | 300 XLA_Shape result_layout; member
|
| /external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| D | collective_lowering.cc | 297 StatusOr<Layout> result_layout = in ConvertBoolReduce() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1732 xla::LayoutProto result_layout = in ExportXlaOp() local
|