Searched refs:ref_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/ |
| D | scatter_nd_ops_test.py | 86 for ref_shape, indices_shape in zip(ref_shapes, indices_shapes):
92 indexable_area_shape += (ref_shape[i],)
109 for i in range(ixdim, len(ref_shape)):
110 updates_shape += (ref_shape[i],)
112 ref = _AsType(np.random.randn(*(ref_shape)), vtype)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 83 for ref_shape, indices_shape in zip(ref_shapes, indices_shapes):
89 indexable_area_shape += (ref_shape[i],)
106 for i in range(ixdim, len(ref_shape)):
107 updates_shape += (ref_shape[i],)
111 np_out = np_scatter(indices, updates, ref_shape)
113 tf_out = tf_scatter(indices, updates, ref_shape)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 713 ref_shape = array_ops.shape(ref, out_type=indices.dtype)
716 ref_shape)
718 ref_grad = array_ops.scatter_nd(indices, grad, ref_shape)
726 ref_shape = gen_resource_variable_ops.variable_shape(ref, indices.dtype)
729 ref_shape)
731 ref_grad = array_ops.scatter_nd(indices, grad, ref_shape)
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 120 for ref_shape, indices_shape in zip(ref_shapes, indices_shapes):
126 indexable_area_shape += (ref_shape[i],)
144 for i in range(ixdim, len(ref_shape)):
145 updates_shape += (ref_shape[i],)
147 ref = _AsType(np.random.randn(*(ref_shape)), vtype)
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 191 auto ref_shape = Shape(scope, ref); in GatherNdGrad() local
193 grad_outputs->push_back(ScatterNd(scope, indices, grad_inputs[0], ref_shape)); in GatherNdGrad()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | functional.py | 618 ref_shape = ref_input.shape
619 ref_rank = ref_shape.rank
628 elif (t_rank == ref_rank - 1 and ref_shape[-1] == 1):
|
| D | base_layer_v1.py | 1295 ref_shape = param.shape
1296 if not ref_shape.is_compatible_with(weight.shape):
1299 'shape %s' % (ref_shape, weight.shape))
|
| D | base_layer.py | 1881 ref_shape = param.shape
1882 if not ref_shape.is_compatible_with(weight.shape):
1885 'shape %s' % (ref_shape, weight.shape))
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | wrappers_test.py | 564 ref_shape = model.layers[-1].output.shape
567 self.assertListEqual(shape.as_list(), ref_shape.as_list())
1134 ref_shape = model.layers[-1].output.shape
1136 self.assertListEqual(shape.as_list(), ref_shape.as_list())
|