Searched defs:tensor_array (Results 1 – 5 of 5) sorted by relevance
75 Status GetTensorArray(OpKernelContext* ctx, TensorArray** tensor_array) { in GetTensorArray()206 TensorArray* tensor_array = new TensorArray( in CreateTensorArray() local308 TensorArray* tensor_array; in CreateTensorArray() local434 TensorArray* tensor_array = nullptr; in Compute() local517 TensorArray* tensor_array = nullptr; in Compute() local605 TensorArray* tensor_array = nullptr; in Compute() local832 TensorArray* tensor_array = nullptr; in Compute() local1041 TensorArray* tensor_array = nullptr; in Compute() local1239 TensorArray* tensor_array = nullptr; in Compute() local1407 TensorArray* tensor_array; in Compute() local[all …]
28 namespace tensor_array { namespace
243 def _ta_append(tensor_array, value): argument
805 def _stack_and_transpose(tensor_array): argument
893 ops::TensorArray tensor_array(root.WithOpName("test/tensor_array"), 1, in TEST() local