Searched defs:partition_ids_shape (Results 1 – 3 of 3) sorted by relevance
48 ShapeHandle partition_ids_shape; in __anon295dc3ce0102() local120 ShapeHandle partition_ids_shape; in __anon295dc3ce0202() local191 ShapeHandle partition_ids_shape; in __anon295dc3ce0302() local
71 ShapeHandle partition_ids_shape; in __anone13797730202() local159 ShapeHandle partition_ids_shape; in __anone13797730402() local289 ShapeHandle partition_ids_shape; in __anone13797730702() local378 ShapeHandle partition_ids_shape; in __anone13797730902() local
228 const TensorShape& partition_ids_shape = partition_ids_t.shape(); in AddToScalarAccumulator() local273 const TensorShape& partition_ids_shape = partition_ids_t.shape(); in AddToTensorAccumulator() local