Searched refs:bcast_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape_test.py | 304 bcast_shape = RaggedTensorDynamicShape.from_dim_sizes(broadcast_dim_sizes)
305 self.assertEqual(original_shape.rank, bcast_shape.rank)
309 bcast2 = bcast_shape.broadcast_dimension(axis, row_length)
311 bcast3 = bcast_shape.broadcast_dimension(axis, 1)
313 self.assertShapeEq(bcast1, bcast_shape)
314 self.assertShapeEq(bcast2, bcast_shape)
315 self.assertShapeEq(bcast3, bcast_shape)
|
| D | ragged_dispatch.py | 203 bcast_shape = ragged_tensor_shape.broadcast_dynamic_shape(
207 x, bcast_shape, broadcast_inner_dimensions=False)
209 y, bcast_shape, broadcast_inner_dimensions=False)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 197 bcast_shape = bcast_batch_shape.concatenate([1, 1])
200 batch_matrices[i] = _broadcast_to_shape(mat, bcast_shape)
209 bcast_shape = array_ops.concat([bcast_batch_shape, [1, 1]], axis=0)
211 batch_matrices[i] = _broadcast_to_shape(mat, bcast_shape)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 51 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compile() local
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | distribution_util_test.py | 311 bcast_shape = sess.run(
314 self.assertAllEqual([2, 5, 3], bcast_shape)
|