Home
last modified time | relevance | path

Searched defs:out_tensor (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dsummary_op_test.cc68 Tensor* out_tensor = GetOutput(0); in TEST_F() local
88 Tensor* out_tensor = GetOutput(0); in TEST_F() local
108 Tensor* out_tensor = GetOutput(0); in TEST_F() local
177 Tensor* out_tensor = GetOutput(0); in TEST_F() local
205 Tensor* out_tensor = GetOutput(0); in TEST_F() local
234 Tensor* out_tensor = GetOutput(0); in TEST_F() local
308 Tensor* out_tensor = GetOutput(0); in TEST_F() local
342 Tensor* out_tensor = GetOutput(0); in TEST_F() local
Dsummary_image_op_test.cc87 Tensor* out_tensor = GetOutput(0); in TEST_F() local
110 Tensor* out_tensor = GetOutput(0); in TEST_F() local
142 Tensor* out_tensor = GetOutput(0); in TEST_F() local
Dsummary_audio_op_test.cc93 Tensor* out_tensor = GetOutput(0); in TEST_F() local
127 Tensor* out_tensor = GetOutput(0); in TEST_F() local
Dsummary_tensor_op_test.cc80 Tensor* out_tensor = GetOutput(0); in TEST_F() local
Dcount_up_to_op.cc49 Tensor* out_tensor; in Compute() local
Dhistogram_op.cc110 Tensor* out_tensor; in Compute() local
Dmkl_transpose_op.cc125 Tensor* out_tensor, const gtl::ArraySlice<int32>& perm) { in MKLTransposeND()
Dfifo_queue.cc91 PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
Dpriority_queue.cc119 OpKernelContext* ctx, PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
Dpadding_fifo_queue.cc60 PersistentTensor* out_tensor) { in GetElementComponent()
Dlist_kernels.h866 Tensor out_tensor; in TensorListBinaryAdd() local
881 Tensor out_tensor; in TensorListZerosLike() local
Drandom_shuffle_queue_op.cc167 PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc111 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
129 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
174 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_from_protos_test.py37 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/
Dinterpreter_utils.cc67 const TfLiteTensor* out_tensor = interpreter->tensor(t); in InterpreterInvokeWithOpResolver() local
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc89 const string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor, in GetTensor()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc29 Tensor* out_tensor) { in ParseTensorProtoToTensor()
625 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
630 Status InMemoryRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
686 TensorProto* out_tensor) { in RecvValue()
691 Status OwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
757 TensorProto* out_tensor) { in RecvValue()
762 Status NonOwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.h146 Tensor out_tensor; in OptionalBinaryAdd() local
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show.cc59 std::unique_ptr<Tensor> out_tensor; in LookUpCheckPoint() local
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py911 def _skip_tensor(self, op_id, out_tensor, report_handler): argument
1533 def conditional_trace_fn(predicate_tensor, out_tensor, trace_fn, argument
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc491 Status PyArrayToTF_Tensor(PyObject* ndarray, Safe_TF_TensorPtr* out_tensor) { in PyArrayToTF_Tensor()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc293 Tensor** out_tensor) { in allocate_persistent()
748 DataType type, const TensorShape& shape, Tensor* out_tensor, in allocate_tensor()
855 Tensor** out_tensor, in allocate_persistent()
Dop_kernel.h1046 Tensor** out_tensor) { in allocate_persistent()
1304 Tensor* out_tensor, in allocate_tensor()
Dop_compatibility_test.cc36 Tensor* out_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc2185 TfLiteTensor* out_tensor = &context->tensors[node->outputs->data[0]]; in CancelOpRegistration() local
2205 TfLiteTensor* out_tensor = &context->tensors[node->outputs->data[0]]; in OkOpRegistration() local

12