Searched refs:predicate_shape (Results 1 – 2 of 2) sorted by relevance
191 Shape predicate_shape = ShapeUtil::MakeShape( in GetExpandedFilterMask() local194 predicate_shape, broadcasted_mask1, broadcasted_mask2, in GetExpandedFilterMask()
2525 const TensorShapeProto& predicate_shape = input_props[0].shape(); in SimplifySelect() local2527 !predicate_shape.unknown_rank() && predicate_shape.dim_size() == 0; in SimplifySelect()