Home
last modified time | relevance | path

Searched defs:out_tensor (Results 1 – 25 of 29) 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.cc48 Tensor* out_tensor; in Compute() local
Dhistogram_op.cc106 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()
Drandom_shuffle_queue_op.cc167 PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
Dlist_kernels.h887 Tensor out_tensor; in TensorListBinaryAdd() local
902 Tensor out_tensor; in TensorListZerosLike() 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/flex/
Dbuffer_map_test.cc117 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
135 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
180 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc90 const string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor, in GetTensor()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc27 Tensor* out_tensor) { in ParseTensorProtoToTensor()
602 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
607 Status InMemoryRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
663 TensorProto* out_tensor) { in RecvValue()
668 Status OwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
734 TensorProto* out_tensor) { in RecvValue()
739 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/contrib/framework/kernels/
Dzero_initializer_op.cc50 Tensor* out_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show.cc58 std::unique_ptr<Tensor> out_tensor; in LookUpCheckPoint() local
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc416 Status PyArrayToTF_Tensor(PyObject* ndarray, Safe_TF_TensorPtr* out_tensor) { in PyArrayToTF_Tensor()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc256 Tensor** out_tensor) { in allocate_persistent()
703 DataType type, const TensorShape& shape, Tensor* out_tensor, in allocate_tensor()
761 Tensor** out_tensor, in allocate_persistent()
Dop_kernel.h993 Tensor** out_tensor) { in allocate_persistent()
1240 Tensor* out_tensor, in allocate_tensor()
Dop_compatibility_test.cc35 Tensor* out_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py940 def _AssertShapesMatch(op_name, in_tensor, out_tensor): argument
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc1510 TfLiteTensor* out_tensor = &context->tensors[node->outputs->data[0]]; in CancelOpRegistration() local
1530 TfLiteTensor* out_tensor = &context->tensors[node->outputs->data[0]]; in OkOpRegistration() local

12