Searched defs:dim0_value (Results 1 – 2 of 2) sorted by relevance
1219 auto dim0_value = dim0->scalar<int32>()(); in SparseSegmentReductionGradShapeFn() local1261 auto dim0_value = dim0->scalar<int32>()(); in SparseSegmentReductionWithNumSegmentsShapeFn() local
75 bool validate_zero_padding(unsigned int dim0_value, unsigned int dim1_value) in validate_zero_padding()