Home
last modified time | relevance | path

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

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