| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 83 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply,
154 const ProgramShape& to_apply);
160 const ProgramShape& to_apply_shape);
165 const Shape& operand_shape, const ProgramShape& select_shape,
167 const Shape& init_value_shape, const ProgramShape& scatter_shape);
207 static StatusOr<Shape> InferWhileShape(const ProgramShape& condition,
208 const ProgramShape& body,
214 absl::Span<const ProgramShape> branch_computations,
271 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply);
292 const Shape& updates_shape, const ProgramShape& to_apply_shape,
|
| D | computation_layout.cc | 26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape, in ComputationLayout()
59 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape()
60 ProgramShape program_shape; in ComputeProgramShape()
|
| D | hlo_module_config.cc | 31 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape, in HloModuleConfig()
37 const ProgramShape& program_shape) { in SetDefaultComputationLayout()
|
| D | computation_layout.h | 42 explicit ComputationLayout(const ProgramShape& program_shape,
88 ProgramShape ComputeProgramShape() const;
|
| D | hlo_module_config.h | 45 explicit HloModuleConfig(const ProgramShape& program_shape,
55 void SetDefaultComputationLayout(const ProgramShape& program_shape);
|
| D | shape_inference_test.cc | 64 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape()
99 ProgramShape select_program_shape_;
100 ProgramShape scatter_program_shape_;
317 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F()
347 ProgramShape select_program_shape_fail = in TEST_F()
358 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()
369 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()
380 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()
567 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()
638 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F()
[all …]
|
| D | local_service.cc | 99 const ProgramShape* program_shape) { in CreateExecutionOptions()
125 ProgramShape program_shape(proto.host_program_shape()); in CompileExecutable()
|
| D | service.h | 190 const ProgramShape& program_shape,
224 const ProgramShape& program_shape,
|
| D | compile_only_service.cc | 85 ProgramShape(instance.computation.host_program_shape()), in CompileAheadOfTime()
|
| D | service.cc | 265 const ProgramShape& program_shape, in CreateModuleConfig()
336 const ProgramShape& program_shape, in CreateModuleConfig()
662 ProgramShape{request.computation().host_program_shape()}, in ExecuteGraphParallel()
837 CreateModuleConfig(ProgramShape{arg->computation().host_program_shape()}, in Compile()
1086 ProgramShape program_shape(arg->computation().host_program_shape()); in ComputeConstantGraph()
1135 HloModuleConfig config(ProgramShape{arg->computation().host_program_shape()}); in GetComputationGraphStats()
|
| /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 | hlo_test_base.cc | 63 bool ProgramShapesEqual(const ProgramShape& lhs, const ProgramShape& rhs) { in ProgramShapesEqual()
75 ProgramShape GetProgramShapeWithLayout(const HloModule& module) { in GetProgramShapeWithLayout()
76 ProgramShape program_shape; in GetProgramShapeWithLayout()
|
| 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 | 186 ProgramShape::ProgramShape(const ProgramShapeProto& program_shape_proto) { in ProgramShape() function in xla::ProgramShape
196 ProgramShapeProto ProgramShape::ToProto() const { in ToProto()
208 string ProgramShape::ToString() const { in ToString()
219 std::ostream& operator<<(std::ostream& out, const ProgramShape& program_shape) { in operator <<()
|
| D | layout_util.h | 73 static void SetToDefaultLayout(ProgramShape* program_shape);
91 static void ClearLayout(ProgramShape* program_shape);
127 static bool HasLayout(const ProgramShape& program_shape);
|
| D | shape.h | 220 class ProgramShape {
222 ProgramShape() = default;
225 explicit ProgramShape(const ProgramShapeProto& program_shape_proto);
293 std::ostream& operator<<(std::ostream& out, const ProgramShape& program_shape);
|
| D | shape_test.cc | 143 ProgramShape program_shape; in TEST_F()
160 ProgramShape program_shape_copy(program_shape.ToProto()); in TEST_F()
180 ProgramShape prog = ShapeUtil::MakeProgramShape( in TEST_F()
|
| D | layout_util.cc | 166 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout()
266 /* static */ void LayoutUtil::ClearLayout(ProgramShape* program_shape) { in ClearLayout()
318 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xrt.h | 93 XrtExecutable(const ProgramShape& program_shape, int64 handle,
102 const ProgramShape& program_shape() const;
106 const ProgramShape program_shape_;
|
| D | xrt.cc | 154 ProgramShape program_shape; in CompileForXrt()
177 XrtExecutable::XrtExecutable(const ProgramShape& program_shape, int64 handle, in XrtExecutable()
244 const ProgramShape& XrtExecutable::program_shape() const { in program_shape()
|
| D | numpy_bridge.h | 73 const ProgramShape& shape);
|
| /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;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 119 ASSERT_TRUE(function.ProgramShape() != nullptr); in TEST()
120 const xla::ProgramShape program_shape(*function.ProgramShape()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | show_signature.cc | 54 std::unique_ptr<ProgramShape> shape = in RealMain()
|