| /external/tensorflow/tensorflow/core/util/ |
| D | memmapped_file_system_writer.cc | 43 const auto tensor_data = tensor.tensor_data(); in SaveTensor() local
44 if (tensor_data.empty()) { in SaveTensor()
51 const auto result = output_file_->Append(tensor_data); in SaveTensor()
53 output_file_offset_ += tensor_data.size(); in SaveTensor()
|
| D | stream_executor_util.h | 34 T* ptr = reinterpret_cast<T*>(const_cast<char*>(t.tensor_data().data())); in AsDeviceMemory()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | ring.h | 163 T* buffer = (T*)output->tensor_data().data(); in RingAllreduce()
165 CopyTensorData<Device>((void*)buffer, (void*)input->tensor_data().data(), in RingAllreduce()
166 output->tensor_data().size()); in RingAllreduce()
186 T* segment_recv = (T*)temp->tensor_data().data(); in RingAllreduce()
281 T* buffer = (T*)output->tensor_data().data(); in RingAllgather()
283 (void*)input->tensor_data().data(), in RingAllgather()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | ring.h | 163 T* buffer = (T*)output->tensor_data().data(); in RingAllreduce()
165 CopyTensorData<Device>((void*)buffer, (void*)input->tensor_data().data(), in RingAllreduce()
166 output->tensor_data().size()); in RingAllreduce()
186 T* segment_recv = (T*)temp->tensor_data().data(); in RingAllreduce()
281 T* buffer = (T*)output->tensor_data().data(); in RingAllgather()
283 (void*)input->tensor_data().data(), in RingAllgather()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op.cc | 190 int64 vector_num_elements, const T* tensor_data, in VectorTensorMultiply() argument
196 (static_cast<int32>(tensor_data[i]) - tensor_offset); in VectorTensorMultiply()
204 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
220 const uint8* y_data_ptr = &(tensor_data->value) + i; in VectorTensorMultiply()
271 (static_cast<int32>(tensor_data[i]) - tensor_offset); in VectorTensorMultiply()
332 const T* tensor_data; in Compute() local
339 tensor_data = y_data; in Compute()
346 tensor_data = x_data; in Compute()
351 vector_data, vector_offset, vector_num_elements, tensor_data, in Compute()
|
| D | quantized_add_op.cc | 291 const T* tensor_data, float min_tensor, in VectorTensorAddition() argument
299 tensor_data[i], min_tensor, max_tensor, output_min, output_max); in VectorTensorAddition()
309 const quint8* tensor_data, float min_tensor, in VectorTensorAddition() argument
351 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition()
374 const int64 tensor_value = static_cast<int64>(tensor_data[i]); in VectorTensorAddition()
394 const quint8* tensor_data, float min_tensor, in VectorTensorAddition() argument
434 const int64 tensor_value = static_cast<int64>(tensor_data[i]); in VectorTensorAddition()
518 const T* tensor_data; in Compute() local
527 tensor_data = y_data; in Compute()
536 tensor_data = x_data; in Compute()
[all …]
|
| D | ops_testutil.cc | 54 auto src = output->tensor_data(); in GetOutput()
55 auto dst = managed_output->tensor_data(); in GetOutput()
|
| D | transpose_functor_gpu.cu.cc | 80 const T* p = reinterpret_cast<const T*>(in.tensor_data().data()); in TransposeSimple()
81 T* q = reinterpret_cast<T*>(const_cast<char*>((out->tensor_data().data()))); in TransposeSimple()
107 auto in_data = reinterpret_cast<const T*>(in.tensor_data().data()); in run()
109 reinterpret_cast<T*>(const_cast<char*>(out->tensor_data().data())); in run()
|
| D | transpose_functor_cpu.cc | 40 const T* p = reinterpret_cast<const T*>(in.tensor_data().data()); in TransposeSimple()
41 T* q = reinterpret_cast<T*>(const_cast<char*>((out->tensor_data().data()))); in TransposeSimple()
|
| D | transpose_functor.h | 153 reinterpret_cast<const T*>(in.tensor_data().data()), in TransposeUsingEigen()
156 reinterpret_cast<T*>(const_cast<char*>(out->tensor_data().data())), in TransposeUsingEigen()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util.cc | 36 StringPiece other_data = other.tensor_data(); in DeepCopy()
41 StringPiece tmp_data = tmp.tensor_data(); in DeepCopy()
81 StringPiece to_data = result->tensor_data(); in Concat()
86 StringPiece from_data = tensor.tensor_data(); in Concat()
130 StringPiece from_data = tensor.tensor_data(); in Split()
143 StringPiece to_data = split->tensor_data(); in Split()
163 const_cast<char*>(split.tensor_data().data())); in Split()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | space_to_depth_test.cc | 29 SpaceToDepthOpModel(const TensorData& tensor_data, int block_size) { in SpaceToDepthOpModel() argument
30 input_ = AddInput(tensor_data); in SpaceToDepthOpModel()
31 output_ = AddOutput(tensor_data); in SpaceToDepthOpModel()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 26 FromOpaquePointer(const_cast<char*>(tensor->tensor_data().data())); in FromTensor()
41 return se::DeviceMemoryBase(const_cast<char*>(tensor.tensor_data().data()), in DeviceMemoryFromTensor()
42 tensor.tensor_data().size()); in DeviceMemoryFromTensor()
|
| D | xla_device_context.cc | 117 << reinterpret_cast<const void*>(cpu_tensor->tensor_data().data()) in CopyCPUTensorToDevice()
119 << reinterpret_cast<const void*>(device_tensor->tensor_data().data()) in CopyCPUTensorToDevice()
209 << reinterpret_cast<const void*>(device_tensor->tensor_data().data()) in CopyDeviceTensorToCPU()
211 << reinterpret_cast<const void*>(cpu_tensor->tensor_data().data()) in CopyDeviceTensorToCPU()
|
| D | xla_compilation_cache.cc | 92 arg_values[i].tensor_data() != other.arg_values[i].tensor_data()) { in operator ==()
111 h, Hash64(arg.tensor_data().data(), arg.tensor_data().size())); in operator ()()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 222 const void* tensor_data = TF_TensorData(src); in CopyTF_TensorStringsToPyArray() local
224 const char* limit = static_cast<const char*>(tensor_data) + tensor_size; in CopyTF_TensorStringsToPyArray()
225 DCHECK(tensor_data != nullptr); in CopyTF_TensorStringsToPyArray()
228 const uint64* offsets = static_cast<const uint64*>(tensor_data); in CopyTF_TensorStringsToPyArray()
230 const char* data = static_cast<const char*>(tensor_data) + offsets_size; in CopyTF_TensorStringsToPyArray()
233 (limit - static_cast<const char*>(tensor_data)); in CopyTF_TensorStringsToPyArray()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 572 const void* sendbuff = p->input->tensor_data().data(); in LoopKernelLaunches()
573 void* recvbuff = const_cast<char*>(p->output->tensor_data().data()); in LoopKernelLaunches()
586 void* buf = const_cast<char*>(buf_t->tensor_data().data()); in LoopKernelLaunches()
592 const void* sendbuff = p->input->tensor_data().data(); in LoopKernelLaunches()
594 p->output ? const_cast<char*>(p->output->tensor_data().data()) in LoopKernelLaunches()
602 const void* sendbuff = p->input->tensor_data().data(); in LoopKernelLaunches()
603 void* recvbuff = const_cast<char*>(p->output->tensor_data().data()); in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants.py | 148 tensor_data = {}
153 tensor_data[get_name(tensor.name)] = var.numpy()
173 "data": tensor_data[input_name],
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | ops.cc | 104 std::copy_n(elem.tensor_data().data(), elem.TotalBytes(), in Initializer()
105 const_cast<char*>(t.tensor_data().data()) + offset); in Initializer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 27 memset(const_cast<char*>(tensor.tensor_data().data()), 0, in make_zeros()
28 tensor.tensor_data().size()); in make_zeros()
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | run_tflite_model_op.cc | 114 auto tensor_bytes = input_tensors[i].tensor_data(); in Compute()
130 auto tensor_bytes = output->tensor_data(); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 413 StringPiece tensor_data = tensor.tensor_data(); in FillInputNode() local
415 ConstByteArray(reinterpret_cast<const uint8*>(tensor_data.data()), in FillInputNode()
416 tensor_data.size(), tensor.dtype()); in FillInputNode()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unique_dataset_op.cc | 171 return Hash64(t.tensor_data().data(), t.tensor_data().size()); in operator ()()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test.py | 48 for input_tensor, tensor_data in zip(input_details, input_data):
49 interpreter.set_tensor(input_tensor['index'], tensor_data.numpy())
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib.py | 130 for input_tensor, tensor_data in zip(input_details, input_data):
131 interpreter.set_tensor(input_tensor["index"], tensor_data)
|