Searched refs:ixdim (Results 1 – 3 of 3) sorted by relevance
50 ixdim = indices.shape[-1]51 num_updates = indices.size // ixdim54 for i in range(ixdim, total_nd):58 output_flat = _FlatOuterDims(ref, ixdim + 1)84 ixdim = indices_shape[-1]87 for i in range(ixdim):105 for i in range(ixdim, len(ref_shape)):
63 ixdim = indices.shape[-1]64 num_updates = indices.size // ixdim67 for i in range(ixdim, total_nd):71 output_flat = _FlatOuterDims(ref, ixdim + 1)113 ixdim = indices_shape[-1]116 for i in range(ixdim):135 for i in range(ixdim, len(ref_shape)):
2886 const DimensionHandle ixdim = c->Dim(indices_shape, -1); in ScatterNdShapeHelper() local2890 if (c->ValueKnown(ixdim)) { in ScatterNdShapeHelper()2891 int64 ix = c->Value(ixdim); in ScatterNdShapeHelper()