Searched defs:in_tensor (Results 1 – 6 of 6) sorted by relevance
37 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
125 Status MKLTransposeND(OpKernelContext* context, const Tensor& in_tensor, in MKLTransposeND()
1584 def model(in_tensor): argument1709 def model(in_tensor): argument
146 def func(in_tensor): argument
596 def collective_fn(in_tensor): argument
1407 const TfLiteTensor* in_tensor; in CancelOpRegistration() local1429 const TfLiteTensor* in_tensor; in OkOpRegistration() local