Searched refs:coords_array (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_gather.cc | 28 inline void Gather(const Array& input_array, const Array& coords_array, in Gather() argument
33 const Shape& coords_shape = coords_array.shape(); in Gather()
35 coords_array.GetBuffer<ArrayDataType::kInt32>().data; in Gather()
107 const Array& coords_array = model->GetArray(op->inputs[1]); in Run() local
108 CHECK(coords_array.data_type == ArrayDataType::kInt32) in Run()
124 Gather<ArrayDataType::kFloat>(input_array, coords_array, &output_array); in Run()
127 Gather<ArrayDataType::kUint8>(input_array, coords_array, &output_array); in Run()
130 Gather<ArrayDataType::kInt32>(input_array, coords_array, &output_array); in Run()
133 Gather<ArrayDataType::kInt64>(input_array, coords_array, &output_array); in Run()
136 Gather<ArrayDataType::kComplex64>(input_array, coords_array, in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tpu_client.cc | 160 std::array<int, 3> coords_array = {coords.x, coords.y, coords.z}; in GetTpuDevices() local
166 core, std::move(local_device_state), task_id, coords_array, in GetTpuDevices()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 55 std::array<int, 3> coords_array = {coord.x(), coord.y(), coord.z()}; in GetTpuDevices() local
59 core.id(), task_id, coords_array, core.core_on_chip_index()); in GetTpuDevices()
|