Home
last modified time | relevance | path

Searched defs:TensorValue (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc151 Status KernelAndDeviceOp::Run(const gtl::InlinedVector<TensorValue, 4>& inputs, in Run()
163 const gtl::InlinedVector<TensorValue, 4>& inputs, in Run()
207 const gtl::InlinedVector<TensorValue, 4>& inputs, in Run()
306 const gtl::InlinedVector<TensorValue, 4>& inputs, in Run()
Dtensor_handle.cc151 Status TensorHandle::TensorValue(tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::TensorHandle
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc179 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
221 const OpKernel& kernel, const gtl::InlinedVector<TensorValue, 4>& inputs) { in CheckOpKernelInput()
231 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput()
Ddataset_test_base.h163 Status AddDatasetInputFromArray(gtl::InlinedVector<TensorValue, 4>* inputs, in AddDatasetInputFromArray()
Dconcatenate_dataset_op_test.cc61 gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateConcatenateDatasetContext()
Dzip_dataset_op_test.cc71 gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateZipDatasetContext()
Dsparse_tensor_slice_dataset_op_test.cc51 OpKernel *const op_kernel, gtl::InlinedVector<TensorValue, 4> *inputs, in CreateSparseTensorSliceDatasetContext()
Drepeat_dataset_op_test.cc53 OpKernel *op_kernel, gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateRepeatDatasetContext()
Dtensor_dataset_op_test.cc57 gtl::InlinedVector<TensorValue, 4> *inputs, in CreateTensorDatasetContext()
Dtake_dataset_op_test.cc50 OpKernel *op_kernel, gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateTakeDatasetContext()
Dtensor_slice_dataset_op_test.cc59 gtl::InlinedVector<TensorValue, 4> *inputs, in CreateTensorSliceDatasetContext()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h513 struct TensorValue { struct
514 TensorValue() : mutex_if_ref(nullptr), tensor(nullptr) {} in TensorValue() argument
515 TensorValue(Tensor* t) // NOLINT(runtime/explicit) in TensorValue() argument
517 TensorValue(mutex* mu, Tensor* t) : mutex_if_ref(mu), tensor(t) {} in TensorValue() argument
519 bool is_ref() const { return mutex_if_ref != nullptr; } in is_ref()
521 mutex* mutex_if_ref; // nullptr if not a ref, != nullptr if a ref
522 Tensor* tensor;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc802 Status InMemoryRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::InMemoryRunStepResponse
847 Status OwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::OwnedProtoRunStepResponse
899 Status NonOwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::NonOwnedProtoRunStepResponse
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc89 gtl::InlinedVector<TensorValue, 4>* inputs, in PrepOp()