| /external/tensorflow/tensorflow/compiler/xla/client/ |
| compile_only_client.h | 44 std::vector<const Shape*> argument_layouts; member in struct:xla::CompileOnlyClient::AotXlaComputationInstance
|
| compile_only_client.cc | 37 service_instance.argument_layouts = instance.argument_layouts;
|
| local_client.h | 121 const absl::Span<const Shape* const> argument_layouts,
|
| local_client.cc | 246 const absl::Span<const Shape* const> argument_layouts,
256 computation, argument_layouts, updated_options));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| local_service.h | 51 const absl::Span<const Shape* const> argument_layouts,
|
| local_service.cc | 121 const absl::Span<const Shape* const> argument_layouts,
128 if (argument_layouts.size() != program_shape.parameters_size()) {
131 program_shape.parameters_size(), argument_layouts.size());
134 for (int i = 0; i < argument_layouts.size(); ++i) {
135 const Shape& argument_shape = *argument_layouts[i];
169 CreateModuleConfig(program_shape, argument_layouts, &execution_options));
|
| compile_only_service.h | 44 std::vector<const Shape*> argument_layouts; member in struct:xla::CompileOnlyService::AotXlaComputationInstance
|
| compile_only_service.cc | 86 instance.argument_layouts, &execution_options));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_test_base.cc | 186 std::vector<const Shape*> argument_layouts(arguments.size());
188 argument_layouts[i] = &arguments[i]->on_host_shape();
192 local_client_->Compile(computation, argument_layouts, build_options));
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| xrt_compile_ops.cc | 114 std::vector<xla::Shape> argument_layouts(
119 argument_layouts[i] = xla::Shape(config.program_shape().parameters(i));
120 argument_layout_ptrs[i] = &argument_layouts[i];
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| replay_computation.cc | 96 std::vector<Shape> argument_layouts; local
97 argument_layouts.reserve(
102 argument_layouts.push_back(Shape(param));
103 argument_layout_ptrs.push_back(&argument_layouts.back());
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compilation_cache.cc | 146 std::vector<const xla::Shape*> argument_layouts(
149 argument_layouts[i] = &result.xla_input_shapes[i];
159 client_->Compile(*result.computation, argument_layouts, build_options);
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| compile.cc | 62 // AotXlaComputationInstance::argument_layouts is a vector of Shape
73 instance.argument_layouts = std::move(arg_layout_ptrs);
|
