Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dsummary_op_test.cc72 Tensor* out_tensor = GetOutput(0); in TEST_F() local
100 Tensor* out_tensor = GetOutput(0); in TEST_F() local
129 Tensor* out_tensor = GetOutput(0); in TEST_F() local
203 Tensor* out_tensor = GetOutput(0); in TEST_F() local
237 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
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()
Dmap_kernels.h233 Tensor out_tensor; in TensorMapBinaryAdd() local
Dlist_kernels.h864 Tensor out_tensor; in TensorListBinaryAdd() local
879 Tensor out_tensor; in TensorListZerosLike() local
Drandom_shuffle_queue_op.cc168 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.cc70 const TfLiteTensor* out_tensor = interpreter->tensor(t); in InterpreterInvokeWithOpResolver() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc27 Tensor* out_tensor) { in ParseTensorProtoToTensor()
621 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
626 Status InMemoryRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
682 TensorProto* out_tensor) { in RecvValue()
687 Status OwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
753 TensorProto* out_tensor) { in RecvValue()
758 Status NonOwnedProtoRunGraphResponse::RecvValue(size_t i, Tensor* out_tensor) { in RecvValue()
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc89 const string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor, in GetTensor()
/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/core/kernels/mkl/
Dmkl_transpose_op.cc126 Tensor* out_tensor, const gtl::ArraySlice<int32>& perm) { in MKLTransposeND()
/external/tensorflow/tensorflow/lite/kernels/
Dlsh_projection.cc177 TfLiteTensor* out_tensor; in Eval() local
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc214 auto& out_tensor = result->dense_tensors[d]; in FastParseSerializedExample() local
361 auto& out_tensor = result->dense_tensors[d]; in FastParseSerializedExample() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc306 Tensor** out_tensor) { in allocate_persistent()
707 DataType type, const TensorShape& shape, Tensor* out_tensor, in allocate_tensor()
816 Tensor** out_tensor, in allocate_persistent()
Dop_kernel.h1037 Tensor** out_tensor) { in allocate_persistent()
1277 Tensor* out_tensor, in allocate_tensor()
Dop_compatibility_test.cc36 Tensor* out_tensor = nullptr; in Compute() local

12