Searched defs:in_tensor (Results 1 – 5 of 5) sorted by relevance
37 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
639 def _NchwVectCToNchw(in_tensor): argument649 def _OihwVectIToHwio(in_tensor): argument659 def _NchwToNchwVectC(in_tensor): argument666 def _HwioToOihw(in_tensor): argument
124 Status MKLTransposeND(OpKernelContext* context, const Tensor& in_tensor, in MKLTransposeND()
940 def _AssertShapesMatch(op_name, in_tensor, out_tensor): argument
1509 TfLiteTensor* in_tensor = &context->tensors[node->inputs->data[0]]; in CancelOpRegistration() local1529 TfLiteTensor* in_tensor = &context->tensors[node->inputs->data[0]]; in OkOpRegistration() local