| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | local_service.cc | 125 const absl::Span<const Shape* const> argument_layouts, in CompileExecutables() argument
132 if (argument_layouts.size() != program_shape.parameters_size()) { in CompileExecutables()
135 program_shape.parameters_size(), argument_layouts.size()); in CompileExecutables()
138 for (int i = 0; i < argument_layouts.size(); ++i) { in CompileExecutables()
139 const Shape& argument_shape = *argument_layouts[i]; in CompileExecutables()
173 CreateModuleConfig(program_shape, argument_layouts, &execution_options)); in CompileExecutables()
|
| D | local_service.h | 52 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 | 103 instance.argument_layouts, &execution_options, &options)); in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | compile_only_client.cc | 46 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.h | 122 const absl::Span<const Shape* const> argument_layouts,
|
| D | local_client.cc | 342 const absl::Span<const Shape* const> argument_layouts, in Compile() argument
352 computation, argument_layouts, updated_options)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 192 std::vector<const Shape*> argument_layouts(arguments.size()); in ExecuteLocally() local
194 argument_layouts[i] = &arguments[i]->on_host_shape(); in ExecuteLocally()
198 local_client_->Compile(computation, argument_layouts, build_options)); in ExecuteLocally()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 113 std::vector<xla::Shape> argument_layouts( in Compile() local
118 argument_layouts[i] = xla::Shape(config.program_shape().parameters(i)); in Compile()
119 argument_layout_ptrs[i] = &argument_layouts[i]; in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_client.cc | 966 absl::optional<std::vector<Shape>> argument_layouts, in CompileForDevices() argument
1006 return Compile(computation, std::move(argument_layouts), build_options, in CompileForDevices()
1012 absl::optional<std::vector<Shape>> argument_layouts, in Compile() argument
1052 if (!argument_layouts) { in Compile()
1055 argument_layouts = program_shape.parameters(); in Compile()
1056 for (Shape& shape : *argument_layouts) { in Compile()
1061 argument_layout_pointers.reserve(argument_layouts->size()); in Compile()
1079 for (Shape& layout : *argument_layouts) { in Compile()
|
| D | local_client.h | 293 absl::optional<std::vector<Shape>> argument_layouts,
303 absl::optional<std::vector<Shape>> argument_layouts,
|
| D | xla_client.py | 153 compile_options.argument_layouts,
524 self.argument_layouts = None
588 compile_options.argument_layouts = argument_shapes
|
| D | xla_client_test.py | 1995 options.argument_layouts = [
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 117 std::vector<Shape> argument_layouts; in CompileExecutable() local
118 argument_layouts.reserve( in CompileExecutable()
123 argument_layouts.push_back(Shape(param)); in CompileExecutable()
124 argument_layout_ptrs.push_back(&argument_layouts.back()); in CompileExecutable()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 153 std::vector<const xla::Shape*> argument_layouts( in BuildExecutable() local
156 argument_layouts[i] = &result.xla_input_shapes[i]; in BuildExecutable()
168 client_->Compile(*result.computation, argument_layouts, build_options)); in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.py | 104 compile_options.argument_layouts,
|
| D | tpu_client.h | 263 absl::optional<std::vector<Shape>> argument_layouts,
272 absl::optional<std::vector<Shape>> argument_layouts,
|
| D | tpu_client.cc | 723 absl::optional<std::vector<Shape>> argument_layouts, in CompileForDevices() argument
763 return Compile(computation, std::move(argument_layouts), build_options, in CompileForDevices()
769 absl::optional<std::vector<Shape>> argument_layouts, in Compile() argument
806 CHECK(!argument_layouts.has_value()); in Compile()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 76 instance.argument_layouts = std::move(arg_layout_ptrs); in CompileXla()
|