Searched refs:InDimBroadcast (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 418 StatusOr<XlaOp> XlaBuilder::InDimBroadcast( in InDimBroadcast() function in xla::XlaBuilder
444 return InDimBroadcast(broadcast_shape, operand, {}); in AddBroadcastSequence()
467 return InDimBroadcast(broadcast_shape, reshaped_operand, in AddBroadcastSequence()
532 InDimBroadcast(broadcasted_shape, from, broadcast_dimensions)); in BinaryOp()
667 return InDimBroadcast(shape, operand, dimensions); in Broadcast()
701 InDimBroadcast(in_dim_shape, operand, broadcast_dimensions)); in BroadcastInDim()
1767 InDimBroadcast(output_shape, new_operand, {})); in Map()
|
| D | xla_builder.h | 613 StatusOr<XlaOp> InDimBroadcast(const Shape& shape, const XlaOp& operand,
|