| /external/tensorflow/tensorflow/core/util/ |
| memmapped_file_system_writer.cc | 43 const auto tensor_data = tensor.tensor_data(); local
44 if (tensor_data.empty()) {
51 const auto result = output_file_->Append(tensor_data);
53 output_file_offset_ += tensor_data.size();
|
| stream_executor_util.h | 34 T* ptr = reinterpret_cast<T*>(const_cast<char*>(t.tensor_data().data()));
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_util.cc | 30 StringPiece other_data = other.tensor_data();
35 StringPiece tmp_data = tmp.tensor_data();
75 StringPiece to_data = result->tensor_data();
80 StringPiece from_data = tensor.tensor_data();
124 StringPiece from_data = tensor.tensor_data();
137 StringPiece to_data = split->tensor_data();
157 const_cast<char*>(split.tensor_data().data()));
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| space_to_depth_test.cc | 29 SpaceToDepthOpModel(const TensorData& tensor_data, int block_size) {
30 input_ = AddInput(tensor_data);
31 output_ = AddOutput(tensor_data);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| ops_testutil.cc | 54 auto src = output->tensor_data();
55 auto dst = managed_output->tensor_data();
|
| quantized_mul_op.cc | 191 int64 vector_num_elements, const T* tensor_data,
197 (static_cast<int32>(tensor_data[i]) - tensor_offset);
205 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements,
221 const uint8* y_data_ptr = &(tensor_data->value) + i;
272 (static_cast<int32>(tensor_data[i]) - tensor_offset);
333 const T* tensor_data; variable
340 tensor_data = y_data;
347 tensor_data = x_data;
352 vector_data, vector_offset, vector_num_elements, tensor_data,
|
| transpose_functor_gpu.cu.cc | 80 const T* p = reinterpret_cast<const T*>(in.tensor_data().data());
81 T* q = reinterpret_cast<T*>(const_cast<char*>((out->tensor_data().data())));
107 auto in_data = reinterpret_cast<const T*>(in.tensor_data().data());
109 reinterpret_cast<T*>(const_cast<char*>(out->tensor_data().data()));
|
| quantized_add_op.cc | 292 const T* tensor_data, float min_tensor,
300 tensor_data[i], min_tensor, max_tensor, output_min, output_max);
310 const quint8* tensor_data, float min_tensor,
352 const uint8* tensor_ptr = &(tensor_data->value) + i;
375 const int64 tensor_value = static_cast<int64>(tensor_data[i]);
395 const quint8* tensor_data, float min_tensor,
435 const int64 tensor_value = static_cast<int64>(tensor_data[i]);
519 const T* tensor_data; variable
528 tensor_data = y_data;
537 tensor_data = x_data
[all...] |
| transpose_functor.h | 153 reinterpret_cast<const T*>(in.tensor_data().data()),
156 reinterpret_cast<T*>(const_cast<char*>(out->tensor_data().data())),
|
| transpose_functor_cpu.cc | 40 const T* p = reinterpret_cast<const T*>(in.tensor_data().data());
41 T* q = reinterpret_cast<T*>(const_cast<char*>((out->tensor_data().data())));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_device_context.cc | 63 << reinterpret_cast<const void*>(cpu_tensor->tensor_data().data())
66 device_tensor->tensor_data().data())
100 device_tensor->tensor_data().data())
102 << reinterpret_cast<const void*>(cpu_tensor->tensor_data().data())
|
| xla_compilation_cache.cc | 70 if (arg_values[i].tensor_data() != other.arg_values[i].tensor_data()) {
89 h, Hash64(arg.tensor_data().data(), arg.tensor_data().size()));
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| ring.h | 163 T* buffer = (T*)output->tensor_data().data();
165 CopyTensorData<Device>((void*)buffer, (void*)input->tensor_data().data(),
166 output->tensor_data().size());
186 T* segment_recv = (T*)temp->tensor_data().data();
281 T* buffer = (T*)output->tensor_data().data();
283 (void*)input->tensor_data().data(),
|
| /external/tensorflow/tensorflow/cc/framework/ |
| ops.cc | 104 std::copy_n(elem.tensor_data().data(), elem.TotalBytes(),
105 const_cast<char*>(t.tensor_data().data()) + offset);
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| ndarray_tensor.cc | 215 const void* tensor_data = TF_TensorData(src); local
217 const char* limit = static_cast<const char*>(tensor_data) + tensor_size;
218 DCHECK(tensor_data != nullptr);
221 const uint64* offsets = static_cast<const uint64*>(tensor_data);
223 const char* data = static_cast<const char*>(tensor_data) + offsets_size;
226 (limit - static_cast<const char*>(tensor_data));
|
| py_func.cc | 361 StringPiece p = t.tensor_data();
394 const_cast<char*>(copy->tensor_data().data()), t.dtype(),
424 StringPiece p = t.tensor_data();
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| parallel_check_op.cc | 88 const char* v0 = t0.tensor_data().data();
89 const char* v1 = t1.tensor_data().data();
|
| xla_launch_op.cc | 105 reinterpret_cast<void*>(const_cast<char*>(t.tensor_data().data()));
112 reinterpret_cast<void*>(const_cast<char*>(t->tensor_data().data()));
292 const_cast<char*>(t->tensor_data().data()), t->tensor_data().size());
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| accumulate_n_optimizer.cc | 27 memset(const_cast<char*>(tensor.tensor_data().data()), 0,
28 tensor.tensor_data().size());
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| unique_dataset_op.cc | 162 return Hash64(t.tensor_data().data(), t.tensor_data().size());
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_manager.cc | 476 const void* sendbuff = p->in_t->tensor_data().data();
477 void* recvbuff = const_cast<char*>(p->out_t->tensor_data().data());
486 void* buf = const_cast<char*>(buf_t->tensor_data().data());
492 const void* sendbuff = p->in_t->tensor_data().data();
494 ? const_cast<char*>(p->out_t->tensor_data().data())
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_control_wrapper.cc | 414 StringPiece tensor_data = tensor.tensor_data(); local
416 ConstByteArray(reinterpret_cast<const uint8*>(tensor_data.data()),
417 tensor_data.size(), tensor.dtype());
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| graph_compiler.cc | 194 reinterpret_cast<const XlaExpression*>(tensor->tensor_data().data());
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_tensor_coding.cc | 159 StringPiece tdata = val.tensor_data();
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| tensor_coding.cc | 199 StringPiece buf = t.tensor_data();
|
