Searched defs:in_tensor (Results 1 – 8 of 8) sorted by relevance
33 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
194 for(auto in_tensor : inputs) variable368 void link_input(OperatorId op, TensorId in_tensor) in link_input()
65 Status MKLTransposeND(OpKernelContext* context, const Tensor& in_tensor, in MKLTransposeND()
401 def conv_func(self, in_tensor, filter_tensor): argument
3438 def model(in_tensor): argument3613 def model(in_tensor): argument3637 def model(in_tensor): argument
144 def func(in_tensor): argument
958 def collective_fn(in_tensor): argument1008 def collective_fn(in_tensor): argument
1479 const TfLiteTensor* in_tensor; in CancelOpRegistration() local1501 const TfLiteTensor* in_tensor; in OkOpRegistration() local