Searched refs:ref_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 82 for ref_shape, indices_shape in zip(ref_shapes, indices_shapes):
88 indexable_area_shape += (ref_shape[i],)
105 for i in range(ixdim, len(ref_shape)):
106 updates_shape += (ref_shape[i],)
110 np_out = np_scatter(indices, updates, ref_shape)
112 tf_out = tf_scatter(indices, updates, ref_shape)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scatter_nd_ops_test.py | 111 for ref_shape, indices_shape in zip(ref_shapes, indices_shapes):
117 indexable_area_shape += (ref_shape[i],)
135 for i in range(ixdim, len(ref_shape)):
136 updates_shape += (ref_shape[i],)
138 ref = _AsType(np.random.randn(*(ref_shape)), vtype)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 483 ref_shape = array_ops.shape(ref, out_type=indices.dtype)
486 ref_shape)
488 ref_grad = array_ops.scatter_nd(indices, grad, ref_shape)
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 182 auto ref_shape = Shape(scope, ref); in GatherNdGrad() local
184 grad_outputs->push_back(ScatterNd(scope, indices, grad_inputs[0], ref_shape)); in GatherNdGrad()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | wrappers_test.py | 325 ref_shape = model.layers[-1].output.get_shape()
328 self.assertListEqual(shape.as_list(), ref_shape.as_list())
|