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