Searched defs:input_shape_tensor (Results 1 – 3 of 3) sorted by relevance
214 const Tensor& input_shape_tensor = context->input(0); in Compute() local
963 Tensor input_shape_tensor(DT_INT32, shape2); in BM_AvgPoolBk() local
940 auto input_shape_tensor = rewriter.create<ConstOp>( in matchAndRewrite() local