Searched refs:bcast_shape (Results 1 – 8 of 8) 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 | 217 bcast_shape = ragged_tensor_shape.broadcast_dynamic_shape(
221 x, bcast_shape, broadcast_inner_dimensions=False)
223 y, bcast_shape, broadcast_inner_dimensions=False)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_binomial_op.cc | 362 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
371 OP_REQUIRES(ctx, TensorShapeUtils::EndsWith(output_shape, bcast_shape), in Compute()
479 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
488 OP_REQUIRES(ctx, TensorShapeUtils::EndsWith(output_shape, bcast_shape), in Compute()
|
| D | parameterized_truncated_normal_op.cc | 769 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
782 OP_REQUIRES(ctx, TensorShapeUtils::EndsWith(output_shape, bcast_shape), in Compute()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 346 bcast_shape = array_ops.concat(
348 batch_matrices[i] = array_ops.broadcast_to(mat, bcast_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 391 vector bcast_shape
393 bcast_shape.push_back(1)
396 …%one_reshape = tosa.RESHAPE(%one_const) {new_shape=bcast_shape} : (tensor<%one_const.type>) -> ten…
397 …%zero_reshape = tosa.RESHAPE(%zero_const) {new_shape=bcast_shape} : (tensor<%zero_const.type>) -> …
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1063 Shape bcast_shape = ShapeUtil::ChangeElementType(shape, PRED); in Broadcast() local
1065 bcast_shape, in Broadcast()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1371 SmallVector<int64_t, 4> bcast_shape(input_rank, 1); in convertEluOp() local
|