Home
last modified time | relevance | path

Searched defs:bcast_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_binomial_op.cc362 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
479 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
Dparameterized_truncated_normal_op.cc769 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc1371 SmallVector<int64_t, 4> bcast_shape(input_rank, 1); in convertEluOp() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1063 Shape bcast_shape = ShapeUtil::ChangeElementType(shape, PRED); in Broadcast() local