Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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
H A Dparameterized_truncated_normal_op.cc771 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_gather_broadcast_reorder.cc170 Shape bcast_shape = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Drandom_ops.cc191 std::vector<int64_t> bcast_shape; in BroadcastParameters() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc1327 SmallVector<int64_t> bcast_shape(input_rank, 1); in convertEluOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc624 const TensorShapeProto& bcast_shape) -> bool { in IsBiasSemanticAdd()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc1459 Shape bcast_shape = ShapeUtil::ChangeElementType(shape, PRED); in Broadcast() local