Searched refs:bcast_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape_test.py | 297 bcast_shape = RaggedTensorDynamicShape.from_dim_sizes(broadcast_dim_sizes)
298 self.assertEqual(original_shape.rank, bcast_shape.rank)
302 bcast2 = bcast_shape.broadcast_dimension(axis, row_length)
304 bcast3 = bcast_shape.broadcast_dimension(axis, 1)
306 self.assertShapeEq(bcast1, bcast_shape)
307 self.assertShapeEq(bcast2, bcast_shape)
308 self.assertShapeEq(bcast3, bcast_shape)
|
| D | ragged_dispatch.py | 212 bcast_shape = ragged_tensor_shape.broadcast_dynamic_shape(
216 x, bcast_shape, broadcast_inner_dimensions=False)
218 y, bcast_shape, broadcast_inner_dimensions=False)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 337 bcast_shape = array_ops.concat(
339 batch_matrices[i] = array_ops.broadcast_to(mat, bcast_shape)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_binomial_op.cc | 360 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
369 OP_REQUIRES(ctx, TensorShapeUtils::EndsWith(output_shape, bcast_shape), in Compute()
|