Home
last modified time | relevance | path

Searched refs:broadcast_sizes (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dmaterialize-broadcasts.mlir6 …// CHECK-DAG: %[[MIN_BC:.+]] = "mhlo.broadcast"(%[[MIN]]) {broadcast_sizes = dense<4> : tensor<1xi…
7 …// CHECK-DAG: %[[MAX_BC:.+]] = "mhlo.broadcast"(%[[MAX]]) {broadcast_sizes = dense<4> : tensor<1xi…
Dops.mlir57 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[1, 2]> : tensor<2xi64>} : (tensor<3xi32>) -…
64 // expected-error@+1 {{broadcast_sizes has rank 2 instead of rank 1}}
65 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[[1, 2]]> : tensor<1x2xi64>} : (tensor<3xi32…
72 …ected-error@+1 {{result rank (3) does not match operand rank (1) plus size of broadcast_sizes (1)}}
73 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[2]> : tensor<1xi64>} : (tensor<3xi32>) -> t…
81 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[2]> : tensor<1xi64>} : (tensor<3xi32>) -> t…
89 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[2]> : tensor<1xi64>} : (tensor<3xi32>) -> t…
Dlhlo-legalize-to-linalg.mlir222 broadcast_sizes = dense<[4, 2, 1]> : tensor<3xi64>
239 broadcast_sizes = dense<[4, 2, 1]> : tensor<3xi64>
Dhlo-legalize-to-linalg.mlir389 …%0 = "mhlo.broadcast"(%arg) {broadcast_sizes = dense<[4, 2, 1]> : tensor<3xi64>} : (tensor<f32>) -…
403 …%0 = "mhlo.broadcast"(%arg) {broadcast_sizes = dense<[4, 2, 1]> : tensor<3xi64>} : (tensor<4x?x16x…
Dcanonicalize.mlir1543 broadcast_sizes = dense<[1]> : tensor<1xi64>} : (tensor<128xf32>) -> tensor<1x128xf32>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dmultiple_return_tuple.mlir12 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[1,2,3]> : tensor<3xi64>} : (tensor<4xi32>) …
Dexport.mlir145 …%0 = "mhlo.broadcast"(%arg0) {broadcast_sizes = dense<[1,2,3]> : tensor<3xi64>} : (tensor<4xi32>) …
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc308 std::vector<int64> broadcast_sizes; in EinsumInverseDiagonal() local
313 broadcast_sizes.push_back(x_shape.dimensions(x_dim)); in EinsumInverseDiagonal()
316 broadcast_sizes.push_back( in EinsumInverseDiagonal()
317 broadcast_sizes[first_label - config.begin()]); in EinsumInverseDiagonal()
320 x = BroadcastInDim(x, broadcast_sizes, labels->at(2)); in EinsumInverseDiagonal()
Dprng.cc341 absl::Span<const int64> broadcast_sizes = {}) { in Uint128AddUint64() argument
347 Broadcast(u128_high + one, broadcast_sizes),
348 Broadcast(u128_high, broadcast_sizes));
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h238 const Shape& operand, absl::Span<const int64> broadcast_sizes);
Dshape_inference.cc2836 const Shape& operand, absl::Span<const int64> broadcast_sizes) { in InferBroadcastShape() argument
2838 for (int64 size : broadcast_sizes) { in InferBroadcastShape()
2846 broadcast_sizes.size()); in InferBroadcastShape()
2847 std::copy(broadcast_sizes.begin(), broadcast_sizes.end(), dimensions.begin()); in InferBroadcastShape()
2849 dimensions.begin() + broadcast_sizes.size()); in InferBroadcastShape()
2853 result.set_dynamic_dimension(broadcast_sizes.size() + i, in InferBroadcastShape()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir550 …// CHECK-DAG: %[[ZERO_MAT:.*]] = "mhlo.broadcast"(%[[ZERO]]) {broadcast_sizes = dense<12> : tensor…
573 …// CHECK-DAG: %[[V5:.*]] = "mhlo.broadcast"(%[[V4]]) {broadcast_sizes = dense<[1, 22, 128]> : tens…
575 …// CHECK-DAG: %[[V7:.*]] = "mhlo.broadcast"(%[[V6]]) {broadcast_sizes = dense<[1, 22, 128]> : tens…
577 …// CHECK-DAG: %[[V9:.*]] = "mhlo.broadcast"(%[[V8]]) {broadcast_sizes = dense<[1, 22, 128]> : tens…
579 …// CHECK-DAG: %[[V11:.*]] = "mhlo.broadcast"(%[[V10]]) {broadcast_sizes = dense<[1, 22, 128]> : te…
581 …// CHECK-DAG: %[[V13:.*]] = "mhlo.broadcast"(%[[V12]]) {broadcast_sizes = dense<[1, 22, 128]> : te…
583 …// CHECK-DAG: %[[V15:.*]] = "mhlo.broadcast"(%[[V14]]) {broadcast_sizes = dense<[1, 22, 128]> : te…
585 …// CHECK-DAG: %[[V17:.*]] = "mhlo.broadcast"(%[[V16]]) {broadcast_sizes = dense<[1, 22, 128]> : te…
612 …// CHECK-DAG: %[[V44:.*]] = "mhlo.broadcast"(%[[V41]]) {broadcast_sizes = dense<7> : tensor<1xi64>…
613 …// CHECK-DAG: %[[V45:.*]] = "mhlo.broadcast"(%[[V0]]) {broadcast_sizes = dense<[7, 22, 128]> : ten…
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h453 XlaOp Broadcast(XlaOp operand, absl::Span<const int64> broadcast_sizes);
1059 absl::Span<const int64> broadcast_sizes);
1634 XlaOp Broadcast(XlaOp operand, absl::Span<const int64> broadcast_sizes);
Dxla_builder.cc867 absl::Span<const int64> broadcast_sizes) { in Broadcast() argument
872 ShapeInference::InferBroadcastShape(*operand_shape, broadcast_sizes)); in Broadcast()
4098 XlaOp Broadcast(const XlaOp operand, absl::Span<const int64> broadcast_sizes) { in Broadcast() argument
4099 return operand.builder()->Broadcast(operand, broadcast_sizes); in Broadcast()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td388 I64ElementsAttr:$broadcast_sizes
Dhlo_ops.td809 I64ElementsAttr:$broadcast_sizes
Dhlo_ops_base.td882 `broadcast_sizes` to the dimensions. The current values of the operand are
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc421 unsigned num_prepended_dims = llvm::size(broadcast_op.broadcast_sizes()); in getIndexingMaps()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md371 <b> `Broadcast(operand, broadcast_sizes)` </b>
376 `broadcast_sizes` | `ArraySlice<int64>` | The sizes of the new dimensions
378 The new dimensions are inserted on the left, i.e. if `broadcast_sizes` has
389 `broadcast_sizes` is `{2, 3}`, then the result will be an array with shape
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc193 I64_ELEMENTS_ATTR_TO_VECTOR(broadcast_sizes);
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc698 auto sizes = op.broadcast_sizes(); in Verify()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1831 if (broadcast.broadcast_sizes().size() == 0) { in EmitLoopFusionFromMlir()