Searched refs:InDimBroadcast (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 208 StatusOr<XlaOp> InDimBroadcast(
|
| D | mlir_hlo_builder.cc | 390 StatusOr<XlaOp> MlirHloBuilder::InDimBroadcast( in InDimBroadcast() function in xla::MlirHloBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 550 StatusOr<XlaOp> XlaBuilder::InDimBroadcast( in InDimBroadcast() function in xla::XlaBuilder
577 return InDimBroadcast(broadcast_shape, operand, {}); in AddBroadcastSequence()
612 return InDimBroadcast(broadcast_shape, reshaped_operand, in AddBroadcastSequence()
662 InDimBroadcast(broadcasted_shape, from, broadcast_dimensions)); in BinaryOp()
886 return InDimBroadcast(shape, operand, dimensions); in Broadcast()
928 InDimBroadcast(in_dim_shape, operand, broadcast_dimensions)); in BroadcastInDim()
2152 InDimBroadcast(output_shape, new_operand, {})); in Map()
|
| D | xla_builder.h | 956 virtual StatusOr<XlaOp> InDimBroadcast(
|