| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.h | 477 struct TensorValue { struct
478 TensorValue() : mutex_if_ref(nullptr), tensor(nullptr) {} in TensorValue() argument
479 explicit TensorValue(Tensor* t) : mutex_if_ref(nullptr), tensor(t) {} in TensorValue() function
480 TensorValue(mutex* mu, Tensor* t) : mutex_if_ref(mu), tensor(t) {} in TensorValue() function
482 bool is_ref() const { return mutex_if_ref != nullptr; } in is_ref()
485 DataType dtype() const { in dtype()
497 DataType dtype_safe() const { in dtype_safe()
506 mutex* mutex_if_ref; // nullptr if not a ref, != nullptr if a ref
507 Tensor* tensor;
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/models/ |
| H A D | log_reader.py | 48 class TensorValue: class
75 def pretty_print_tensor_value(tv: TensorValue):
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.cc | 325 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateDatasetContext()
489 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
495 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext()
539 const OpKernel& kernel, const gtl::InlinedVector<TensorValue, 4>& inputs) { in CheckOpKernelInput()
549 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | message_wrappers.cc | 829 Status InMemoryRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::InMemoryRunStepResponse
876 Status OwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::OwnedProtoRunStepResponse
932 Status NonOwnedProtoRunStepResponse::TensorValue(size_t i, in TensorValue() function in tensorflow::NonOwnedProtoRunStepResponse
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | tensor_handle_data.cc | 33 Status LocalTensorHandleData::TensorValue(tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::LocalTensorHandleData
|
| H A D | kernel_and_device.h | 74 explicit EagerKernelArgs(gtl::InlinedVector<TensorValue, 4>&& tensor_args) in EagerKernelArgs()
|
| H A D | tensor_handle.cc | 494 Status TensorHandle::TensorValue(const Device* d, tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::TensorHandle
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | map_defun_op_test.cc | 108 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateMapDefunContext()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scoped_allocator_ops_test.cc | 90 gtl::InlinedVector<TensorValue, 4>* inputs, in PrepOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | eager_service_impl_test.cc | 720 gtl::InlinedVector<TensorValue, 4>&& tensor_args, in TestExecuteNodeArgs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | process_function_library_runtime_test.cc | 866 gtl::InlinedVector<TensorValue, 4>&& tensor_args) { in TestFunctionPackedArgs()
|