| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 83 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply,
162 const ProgramShape& to_apply);
168 const ProgramShape& to_apply_shape);
175 const ProgramShape& to_apply_shape);
184 const Shape& operand_shape, const ProgramShape& select_shape,
186 const Shape& init_value_shape, const ProgramShape& scatter_shape);
226 static StatusOr<Shape> InferWhileShape(const ProgramShape& condition,
227 const ProgramShape& body,
233 absl::Span<const ProgramShape> branch_computations,
300 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply);
[all …]
|
| D | hlo_module_config.cc | 31 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape, in HloModuleConfig()
40 const ProgramShape& program_shape) { in SetDefaultComputationLayout()
45 const ProgramShape& program_shape) { in SetComputationLayoutIfExists()
|
| D | computation_layout.cc | 27 ComputationLayout::ComputationLayout(const ProgramShape& program_shape, in ComputationLayout()
73 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape()
74 ProgramShape program_shape; in ComputeProgramShape()
|
| D | computation_layout.h | 42 explicit ComputationLayout(const ProgramShape& program_shape,
90 ProgramShape ComputeProgramShape() const;
|
| D | hlo_module_config.h | 74 explicit HloModuleConfig(const ProgramShape& program_shape,
87 void SetDefaultComputationLayout(const ProgramShape& program_shape);
91 void SetComputationLayoutIfExists(const ProgramShape& program_shape);
|
| D | shape_inference_test.cc | 66 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape()
101 ProgramShape select_program_shape_;
102 ProgramShape scatter_program_shape_;
329 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F()
359 ProgramShape select_program_shape_fail = in TEST_F()
370 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()
381 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()
392 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()
944 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()
1015 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F()
[all …]
|
| D | hlo_module_util.h | 36 const ProgramShape& program_shape,
|
| D | service.h | 189 const ProgramShape& program_shape,
198 const ProgramShape& program_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | replay_test.cc | 54 std::unique_ptr<ProgramShape> original_shape = in TEST_F()
56 std::unique_ptr<ProgramShape> replayed_shape = in TEST_F()
87 std::unique_ptr<ProgramShape> original_shape = in XLA_TEST_F()
89 std::unique_ptr<ProgramShape> replayed_shape = in XLA_TEST_F()
134 std::unique_ptr<ProgramShape> original_shape = in TEST_F()
136 std::unique_ptr<ProgramShape> replayed_shape = in TEST_F()
|
| D | axpy_simple_test.cc | 65 TF_ASSERT_OK_AND_ASSIGN(ProgramShape shape, builder.GetProgramShape()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 196 ProgramShape::ProgramShape(const ProgramShapeProto& program_shape_proto) { in ProgramShape() function in xla::ProgramShape
206 ProgramShapeProto ProgramShape::ToProto() const { in ToProto()
218 string ProgramShape::ToString() const { in ToString()
229 std::ostream& operator<<(std::ostream& out, const ProgramShape& program_shape) { in operator <<()
|
| D | layout_util.h | 74 static void SetToDefaultLayout(ProgramShape* program_shape);
92 static void ClearLayout(ProgramShape* program_shape);
117 static bool HasLayout(const ProgramShape& program_shape);
|
| D | shape.h | 271 class ProgramShape {
273 ProgramShape() = default;
276 explicit ProgramShape(const ProgramShapeProto& program_shape_proto);
344 std::ostream& operator<<(std::ostream& out, const ProgramShape& program_shape);
|
| D | shape_test.cc | 146 ProgramShape program_shape; in TEST_F()
163 ProgramShape program_shape_copy(program_shape.ToProto()); in TEST_F()
183 ProgramShape prog = ShapeUtil::MakeProgramShape( in TEST_F()
|
| D | layout_util.cc | 168 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout()
268 /* static */ void LayoutUtil::ClearLayout(ProgramShape* program_shape) { in ClearLayout()
307 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_computation.cc | 26 StatusOr<ProgramShape> XlaComputation::GetProgramShape() const { in GetProgramShape()
28 return ProgramShape(proto_.host_program_shape()); in GetProgramShape()
|
| D | xla_computation.h | 46 StatusOr<ProgramShape> GetProgramShape() const;
|
| D | compile_only_client.h | 62 const ProgramShape& program_shape,
|
| D | compile_only_client.cc | 27 const ProgramShape& program_shape, in CreateModuleConfig()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 219 ASSERT_TRUE(function.ProgramShape() != nullptr); in TEST()
220 const xla::ProgramShape program_shape(*function.ProgramShape()); in TEST()
270 ASSERT_TRUE(function.ProgramShape() != nullptr); in TEST()
271 const xla::ProgramShape program_shape(*function.ProgramShape()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 257 py::class_<ProgramShape>(m, "ProgramShape") in BuildXlaCompilerSubmodule()
259 [](absl::Span<const Shape> params, Shape result) -> ProgramShape { in BuildXlaCompilerSubmodule()
260 ProgramShape program_shape; in BuildXlaCompilerSubmodule()
268 static_cast<const std::vector<Shape>& (ProgramShape::*)() const>( in BuildXlaCompilerSubmodule()
269 &ProgramShape::parameters)) in BuildXlaCompilerSubmodule()
270 .def("result_shape", &ProgramShape::result) in BuildXlaCompilerSubmodule()
271 .def("__repr__", &ProgramShape::ToString); in BuildXlaCompilerSubmodule()
398 absl::optional<XlaOp> root) -> StatusOr<ProgramShape> { in BuildXlaCompilerSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_topk_test.cc | 37 TF_ASSERT_OK_AND_ASSIGN(ProgramShape program_shape, in TEST_F()
64 TF_ASSERT_OK_AND_ASSIGN(ProgramShape program_shape, in TEST_F()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.h | 96 const xla::ProgramShape& program_shape,
108 const xla::ProgramShape& program_shape,
|
| D | tpu_compile_op_support.cc | 37 using ::xla::ProgramShape;
57 const ProgramShape& program_shape, absl::Span<const Shape> argument_shapes, in CreateModuleConfig()
135 const xla::ProgramShape& program_shape, in CreateModuleConfig()
320 xla::ProgramShape( in CreateHloModules()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | show_signature.cc | 54 std::unique_ptr<ProgramShape> shape = in RealMain()
|