HomeSort by relevance Sort by last modified time
    Searched refs:ProgramShape (Results 1 - 25 of 45) sorted by null

1 2

  /external/tensorflow/tensorflow/compiler/aot/
compile.h 36 xla::ProgramShape program_shape; // Static shape of args and results.
compile.cc 51 xla::StatusOr<std::unique_ptr<xla::ProgramShape>> pshape_or =
58 xla::ProgramShape* pshape = &compile_result->program_shape;
codegen.cc 103 const xla::ProgramShape& ps = compile_result.program_shape;
153 Status GenArgMethods(const tf2xla::Config& config, const xla::ProgramShape& ps,
193 const xla::ProgramShape& ps, string* methods) {
299 const xla::ProgramShape& ps = compile_result.program_shape;
486 static const xla::ProgramShape* StaticProgramShape() {
487 static const xla::ProgramShape* kShape = {{PROGRAM_SHAPE_SHIM_EXPRESSION}};
539 std::unique_ptr<xla::ProgramShape> program_shape;
543 tensorflow::MakeUnique<xla::ProgramShape>(compile_result.program_shape);
557 "xla::ProgramShape", program_shape.get()));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module_config.cc 33 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape)
37 const ProgramShape& program_shape) {
hlo_module_config.h 39 // ProgramShape creates a computation layout using this shape.
41 explicit HloModuleConfig(const ProgramShape& program_shape);
50 void SetDefaultComputationLayout(const ProgramShape& program_shape);
shape_inference.h 86 const ProgramShape& to_apply,
136 const ProgramShape& to_apply);
142 const ProgramShape& to_apply_shape);
147 const Shape& operand_shape, const ProgramShape& select_shape,
149 const Shape& init_value_shape, const ProgramShape& scatter_shape);
189 static StatusOr<Shape> InferWhileShape(const ProgramShape& condition,
190 const ProgramShape& body,
196 const Shape& false_operand, const ProgramShape& true_computation,
197 const ProgramShape& false_computation);
248 const ProgramShape& to_apply)
    [all...]
computation_layout.cc 26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape)
computation_layout.h 35 // Constructs a ComputationLayout from a ProgramShape. The layouts of the
37 // ProgramShape are ignored.
38 explicit ComputationLayout(const ProgramShape& program_shape);
compile_only_service.cc 96 std::shared_ptr<const ProgramShape> program_shape,
service.h 252 const ProgramShape& program_shape,
277 const ProgramShape& program_shape,
user_computation.h 277 StatusOr<std::shared_ptr<const ProgramShape>> ComputeProgramShape(
402 // Memoized ProgramShape and its version. A shared_ptr is used because
405 mutable std::shared_ptr<const ProgramShape> program_shape_ GUARDED_BY(mutex_);
  /external/tensorflow/tensorflow/compiler/xla/client/
computation.h 65 StatusOr<ProgramShape> GetProgramShape() const;
computation.cc 67 StatusOr<ProgramShape> Computation::GetProgramShape() const {
client.h 157 StatusOr<std::unique_ptr<ProgramShape>> GetComputationShape(
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiled_cpu_function.h 28 class ProgramShape;
78 const xla::ProgramShape* program_shape = nullptr;
208 const xla::ProgramShape* ProgramShape() const { return program_shape_; }
240 const xla::ProgramShape* program_shape_ = nullptr;
xla_jit_compiled_cpu_function.h 82 std::unique_ptr<const xla::ProgramShape> program_shape_;
xla_jit_compiled_cpu_function_test.cc 119 const xla::ProgramShape* program_shape = function.ProgramShape();
xla_jit_compiled_cpu_function.cc 40 const xla::ProgramShape& program_shape) {
123 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::ProgramShape> program_shape,
  /external/tensorflow/tensorflow/compiler/xla/tests/
replay_test.cc 54 std::unique_ptr<ProgramShape> original_shape =
56 std::unique_ptr<ProgramShape> replayed_shape =
86 std::unique_ptr<ProgramShape> original_shape =
88 std::unique_ptr<ProgramShape> replayed_shape =
132 std::unique_ptr<ProgramShape> original_shape =
134 std::unique_ptr<ProgramShape> replayed_shape =
axpy_simple_test.cc 65 TF_ASSERT_OK_AND_ASSIGN(ProgramShape shape, builder.GetProgramShape());
hlo_test_base.cc 63 bool ProgramShapesEqual(const ProgramShape& lhs, const ProgramShape& rhs) {
75 ProgramShape GetProgramShapeWithLayout(const HloModule& module) {
76 ProgramShape program_shape;
  /external/tensorflow/tensorflow/compiler/xla/
layout_util.h 59 // Sets the layouts of all Shapes within the given ProgramShape to the
61 static void SetToDefaultLayout(ProgramShape* program_shape);
75 // Clears the layout on all Shapes within the given ProgramShape.
76 static void ClearLayout(ProgramShape* program_shape);
129 // Returns whether all Shapes within the given ProgramShape have layouts.
130 static bool HasLayout(const ProgramShape& program_shape);
  /external/tensorflow/tensorflow/compiler/xla/tools/
show_signature.cc 18 // Shows the signature (ProgramShape) of binary snapshot proto(s) on the command
56 std::unique_ptr<ProgramShape> shape =
dumped_computation_to_operation_list.cc 80 std::unique_ptr<ProgramShape> program_shape =
dumped_computation_to_text.cc 55 std::unique_ptr<ProgramShape> program_shape =

Completed in 967 milliseconds

1 2