Searched defs:input_shape_tensor (Results 1 – 3 of 3) sorted by relevance
129 def _forward_event_shape_tensor(self, input_shape_tensor): argument
208 const Tensor& input_shape_tensor = context->input(0); in Compute() local
925 Tensor input_shape_tensor(DT_INT32, shape2); in BM_AvgPoolBk() local