Searched defs:pred_input_shape (Results 1 – 1 of 1) sorted by relevance
326 const Shape pred_input_shape = in GenerateBinaryMask() local480 const Shape pred_input_shape = in RewriteDynamicReshapeSplitInput() local