Searched refs:ixdim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/ |
| D | scatter_nd_ops_test.py | 50 ixdim = indices.shape[-1]
51 num_updates = indices.size // ixdim
54 for i in range(ixdim, total_nd):
58 output_flat = _FlatOuterDims(ref, ixdim + 1)
88 ixdim = indices_shape[-1]
91 for i in range(ixdim):
109 for i in range(ixdim, len(ref_shape)):
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 51 ixdim = indices.shape[-1]
52 num_updates = indices.size // ixdim
55 for i in range(ixdim, total_nd):
59 output_flat = _FlatOuterDims(ref, ixdim + 1)
85 ixdim = indices_shape[-1]
88 for i in range(ixdim):
106 for i in range(ixdim, len(ref_shape)):
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 64 ixdim = indices.shape[-1]
65 num_updates = indices.size // ixdim
68 for i in range(ixdim, total_nd):
72 output_flat = _FlatOuterDims(ref, ixdim + 1)
122 ixdim = indices_shape[-1]
125 for i in range(ixdim):
144 for i in range(ixdim, len(ref_shape)):
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2336 const DimensionHandle ixdim = c->Dim(indices_shape, -1); in ScatterNdShapeHelper() local
2340 if (c->ValueKnown(ixdim)) { in ScatterNdShapeHelper()
2341 int64 ix = c->Value(ixdim); in ScatterNdShapeHelper()
|