| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 510 struct TensorValue { struct
511 TensorValue() : mutex_if_ref(nullptr), tensor(nullptr) {} in TensorValue() function
512 explicit TensorValue(Tensor* t) : mutex_if_ref(nullptr), tensor(t) {} in TensorValue() function
513 TensorValue(mutex* mu, Tensor* t) : mutex_if_ref(mu), tensor(t) {} in TensorValue() argument
515 bool is_ref() const { return mutex_if_ref != nullptr; } in is_ref()
518 DataType dtype() const { in dtype()
530 DataType dtype_safe() const { in dtype_safe()
539 mutex* mutex_if_ref; // nullptr if not a ref, != nullptr if a ref
540 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/kernels/data/ |
| D | dataset_test_base.cc | 320 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateDatasetContext()
484 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
490 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
534 const OpKernel& kernel, const gtl::InlinedVector<TensorValue, 4>& inputs) { in CheckOpKernelInput()
544 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput()
|
| 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 | 821 Status InMemoryRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::InMemoryRunStepResponse
866 Status OwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::OwnedProtoRunStepResponse
918 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/common_runtime/ |
| D | process_function_library_runtime_test.cc | 817 gtl::InlinedVector<TensorValue, 4>&& tensor_args) { in TestFunctionPackedArgs()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 714 gtl::InlinedVector<TensorValue, 4>&& tensor_args, in TestExecuteNodeArgs()
|