| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 467 struct TensorValue { struct
468 TensorValue() : mutex_if_ref(nullptr), tensor(nullptr) {} in TensorValue() argument
469 explicit TensorValue(Tensor* t) : mutex_if_ref(nullptr), tensor(t) {} in TensorValue() function
470 TensorValue(mutex* mu, Tensor* t) : mutex_if_ref(mu), tensor(t) {} in TensorValue() function
472 bool is_ref() const { return mutex_if_ref != nullptr; } in is_ref()
475 DataType dtype() const { in dtype()
487 DataType dtype_safe() const { in dtype_safe()
496 mutex* mutex_if_ref; // nullptr if not a ref, != nullptr if a ref
497 Tensor* tensor;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_data.cc | 33 Status LocalTensorHandleData::TensorValue(tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::LocalTensorHandleData
|
| D | kernel_and_device.h | 68 explicit EagerKernelArgs(gtl::InlinedVector<TensorValue, 4>&& tensor_args) in EagerKernelArgs()
|
| D | tensor_handle.cc | 494 Status TensorHandle::TensorValue(const Device* d, tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::TensorHandle
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.cc | 315 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateDatasetContext()
479 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
485 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
529 const OpKernel& kernel, const gtl::InlinedVector<TensorValue, 4>& inputs) { in CheckOpKernelInput()
539 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 108 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateMapDefunContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.cc | 817 Status InMemoryRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::InMemoryRunStepResponse
862 Status OwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::OwnedProtoRunStepResponse
914 Status NonOwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::NonOwnedProtoRunStepResponse
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops_test.cc | 90 gtl::InlinedVector<TensorValue, 4>* inputs, in PrepOp()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 714 gtl::InlinedVector<TensorValue, 4>&& tensor_args, in TestExecuteNodeArgs()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 809 gtl::InlinedVector<TensorValue, 4>&& tensor_args) { in TestFunctionPackedArgs()
|