| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 119 LocalExecutable::RunHelper(const absl::Span<const Shape* const> argument_shapes, in RunHelper() argument
125 const int argument_shapes_size = argument_shapes.size(); in RunHelper()
129 computation_layout.parameter_count(), argument_shapes.size()); in RunHelper()
131 for (int i = 0, end = argument_shapes.size(); i < end; ++i) { in RunHelper()
133 *argument_shapes[i], /*minor_to_major_only=*/true)) { in RunHelper()
142 ShapeUtil::HumanStringWithLayout(*argument_shapes[i])); in RunHelper()
175 std::vector<const Shape*> argument_shapes; in Run() local
176 argument_shapes.reserve(arguments.size()); in Run()
178 argument_shapes.push_back(&arg->on_device_shape()); in Run()
181 argument_shapes, run_options, [&](const ExecutableRunOptions& options) { in Run()
[all …]
|
| D | local_client.h | 93 const absl::Span<const Shape* const> argument_shapes,
103 absl::Span<Shape const* const> argument_shapes, in AsyncCallAndBlockHostUntilDone() argument
107 RunHelper(argument_shapes, run_options)); in AsyncCallAndBlockHostUntilDone()
|
| D | compile_only_client.cc | 28 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig() argument
30 return compiler_service_->CreateModuleConfig(program_shape, argument_shapes, in CreateModuleConfig()
|
| D | compile_only_client.h | 63 absl::Span<const Shape* const> argument_shapes,
|
| D | client.h | 63 absl::Span<const Shape> argument_shapes,
|
| D | client.cc | 218 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile() argument
235 for (const auto& arg_shape : argument_shapes) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_util.cc | 46 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig() argument
52 const int64 argument_shapes_size = argument_shapes.size(); in CreateModuleConfig()
56 argument_shapes.size()); in CreateModuleConfig()
58 for (int i = 0, end = argument_shapes.size(); i < end; ++i) { in CreateModuleConfig()
61 if (!ShapeUtil::Compatible(*argument_shapes[i], in CreateModuleConfig()
67 ShapeUtil::HumanString(*argument_shapes[i])); in CreateModuleConfig()
71 *argument_shapes[i])); in CreateModuleConfig()
|
| D | hlo_module_util.h | 37 absl::Span<const Shape* const> argument_shapes,
|
| D | service.cc | 257 absl::Span<const Shape* const> argument_shapes, in CreateModuleConfig() argument
267 return xla::CreateModuleConfig(program_shape, argument_shapes, in CreateModuleConfig()
277 std::vector<const Shape*> argument_shapes; in CreateModuleConfig() local
279 argument_shapes.push_back(&arg->on_host_shape()); in CreateModuleConfig()
281 return CreateModuleConfig(program_shape, argument_shapes, &execution_options, in CreateModuleConfig()
804 std::vector<Shape> argument_shapes; in Compile() local
805 argument_shapes.reserve(arg->input_shape_with_layout_size()); in Compile()
808 argument_shapes.push_back(Shape(shape_proto)); in Compile()
809 argument_shape_ptrs.push_back(&argument_shapes.back()); in Compile()
|
| D | service.h | 190 absl::Span<const Shape* const> argument_shapes,
|
| D | shape_inference.cc | 3183 string argument_shapes = in InferCallShape() local
3192 argument_shapes); in InferCallShape()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 57 const ProgramShape& program_shape, absl::Span<const Shape> argument_shapes, in CreateModuleConfig() argument
67 if (program_shape.parameters_size() != argument_shapes.size()) { in CreateModuleConfig()
70 argument_shapes.size()); in CreateModuleConfig()
72 for (int i = 0; i < argument_shapes.size(); ++i) { in CreateModuleConfig()
75 if (!ShapeUtil::Compatible(argument_shapes[i], in CreateModuleConfig()
81 ShapeUtil::HumanString(argument_shapes[i])); in CreateModuleConfig()
85 argument_shapes[i])); in CreateModuleConfig()
136 absl::Span<const Shape> argument_shapes, in CreateModuleConfig() argument
140 return CreateModuleConfig(program_shape, argument_shapes, result_layout, in CreateModuleConfig()
|
| D | tpu_compile_op_support.h | 97 absl::Span<const xla::Shape> argument_shapes,
109 absl::Span<const xla::Shape> argument_shapes,
|