| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | fold-broadcast.mlir | 6 %0 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<7xf32>, tensor<2xi32>) -> tensor<5x7xf32>
16 %0 = "tf.BroadcastTo"(%arg0, %cst) : (tensor<7xf32>, tensor<2xi32>) -> tensor<5x7xf32>
26 %0 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<7xf32>, tensor<2xi32>) -> tensor<5x7xf32>
36 %0 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<5xf32>, tensor<3xi32>) -> tensor<3x5x7xf32>
40 …// CHECK: %[[V0:.*]] = "tf.BroadcastTo"(%arg1, %[[C0]]) : (tensor<5xf32>, tensor<3xi32>) -> tensor…
48 %0 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<7xf32>, tensor<2xi32>) -> tensor<5x7xf32>
58 %0 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<7xf32>, tensor<2xi32>) -> tensor<5x7xf32>
68 %0 = "tf.BroadcastTo"(%arg0, %cst) : (tensor<7xf32>, tensor<2xi64>) -> tensor<5x7xf32>
69 %1 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<5x1xf32>, tensor<2xi64>) -> tensor<5x7xf32>
79 %0 = "tf.BroadcastTo"(%arg1, %cst) : (tensor<17x24xf32>, tensor<3xi64>) -> tensor<17x17x24xf32>
[all …]
|
| D | tf_optimize.mlir | 38 …%97 = "tf.BroadcastTo"(%arg0, %cst_1) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor<8xi64>) -> tensor<1…
46 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[RESHAPE]], %[[CST1]]) : (tensor<8x1x1x18xbf16>, …
53 …%97 = "tf.BroadcastTo"(%arg0, %cst_1) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor<9xi64>) -> tensor<7…
61 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[RESHAPE]], %[[CST1]]) : (tensor<1x8x1x1x18xbf16>…
68 …%97 = "tf.BroadcastTo"(%arg0, %cst_1) : (tensor<1x1x1x1x1x1x1x18xbf16>, tensor<8xi64>) -> tensor<1…
76 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[RESHAPE]], %[[CST1]]) : (tensor<1x1x1x1x18xbf16>…
83 …%97 = "tf.BroadcastTo"(%arg0, %cst_1) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor<8xi64>) -> tensor<1…
90 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%arg0, %[[CST]]) : (tensor<1x8x1x1x1x1x1x18xbf16>, …
100 …%97 = "tf.BroadcastTo"(%arg0, %cst_1) : (tensor<1x1x1x1x1x1x1x18xbf16>, tensor<8xi64>) -> tensor<1…
107 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%arg0, %[[CST]]) : (tensor<1x1x1x1x1x1x1x18xbf16>, …
[all …]
|
| D | tensor_list_ops_decomposition.mlir | 14 …// CHECK-NEXT: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[CAST_ZERO]], %[[CONST10]]) : (tensor<f32>, …
56 …// CHECK-NEXT: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[CAST_ZERO]], %[[CONST10]]) : (tensor<f32>, …
362 // CHECK: %[[BUFFER:.*]] = "tf.BroadcastTo"(%[[ZERO_F32]], %[[MAX_SIZE]])
403 // CHECK: %[[BUFFER:.*]] = "tf.BroadcastTo"(%[[ZERO_F32]], %[[MAX_SIZE]])
456 // CHECK: %[[INIT:.*]] = "tf.BroadcastTo"
499 // CHECK: %[[INIT:.*]] = "tf.BroadcastTo"
545 // CHECK: "tf.BroadcastTo"
559 // CHECK: tf.BroadcastTo
563 // CHECK: tf.BroadcastTo
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | beta_op.cc | 67 auto a, BroadcastTo(ctx->Input(0), merged_shape.dim_sizes())); in Compile()
69 auto b, BroadcastTo(ctx->Input(1), merged_shape.dim_sizes())); in Compile()
71 auto x, BroadcastTo(ctx->Input(2), merged_shape.dim_sizes())); in Compile()
|
| D | select_op.cc | 114 auto bcasted_cond = BroadcastTo(ctx->Input(0), bcast.output_shape()); in Compile()
118 auto bcasted_then = BroadcastTo(ctx->Input(1), bcast.output_shape()); in Compile()
122 auto bcasted_else = BroadcastTo(ctx->Input(2), bcast.output_shape()); in Compile()
|
| D | cwise_ops.cc | 91 auto lhs_output = BroadcastTo(lhs, broadcast_helper.output_shape()); in Broadcast()
96 auto rhs_output = BroadcastTo(rhs, broadcast_helper.output_shape()); in Broadcast()
|
| D | random_ops.cc | 343 BroadcastTo(ctx->Input(1), shape.dim_sizes())); in Compile()
345 BroadcastTo(ctx->Input(2), shape.dim_sizes())); in Compile()
347 BroadcastTo(ctx->Input(3), shape.dim_sizes())); in Compile()
349 BroadcastTo(ctx->Input(4), shape.dim_sizes())); in Compile()
|
| D | matrix_triangular_solve_op.cc | 92 auto lhs_output = BroadcastTo(lhs, lhs_broadcast_shape.dim_sizes()); in Broadcast()
101 auto rhs_output = BroadcastTo(rhs, rhs_broadcast_shape.dim_sizes()); in Broadcast()
|
| D | broadcast_to_op.cc | 35 BroadcastTo(context->Input(0), output_shape.dim_sizes()); in Compile()
|
| D | tile_ops.cc | 87 auto result_or = BroadcastTo(ctx->Input("input"), output_dims); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | broadcast_to_op_test.cc | 27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo() function
51 BroadcastTo(DIM0, DIM1, \
66 BroadcastTo(DIM0, DIM1, \
|
| D | broadcast_to_op.cc | 90 functor::BroadcastTo<Device, T>()(device, ctx, *output_tensor, output_shape, in Compute()
111 void BroadcastTo<GPUDevice, Type>::operator()( \
115 extern template struct BroadcastTo<GPUDevice, Type>;
|
| D | broadcast_to_op_gpu.cu.cc | 29 template class functor::BroadcastTo<GPUDevice, Type>;
|
| D | broadcast_to_op.h | 33 struct BroadcastTo { struct
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 31 xla::StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input, in BroadcastTo() function
107 TF_ASSIGN_OR_RETURN(*lhs, BroadcastTo(*lhs, bcast.output_shape())); in BroadcastOpsToSame()
108 TF_ASSIGN_OR_RETURN(*rhs, BroadcastTo(*rhs, bcast.output_shape())); in BroadcastOpsToSame()
|
| D | broadcast.h | 27 xla::StatusOr<xla::XlaOp> BroadcastTo(xla::XlaOp input,
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_test.cc | 32 Node* BroadcastTo(Graph* g, Node* input, Node* shape) { in BroadcastTo() function
82 in0_node = BroadcastTo(g, test::graph::Constant(g, in0), in MatrixTriangularSolveWithBroadcast()
84 in1_node = BroadcastTo(g, test::graph::Constant(g, in1), in MatrixTriangularSolveWithBroadcast()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_BroadcastTo.pbtxt | 2 graph_op_name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BroadcastTo.pbtxt | 2 name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BroadcastTo.pbtxt | 2 name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf-no-runtime-verification.mlir | 4 %0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<3xbf16>, tensor<2xi64>) -> tensor<3x3xbf16>
|
| D | prepare-tf.mlir | 585 %0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<3xf32>, tensor<2xi32>) -> tensor<3x3xf32>
595 %0 = "tf.BroadcastTo"(%input, %shape) : (tensor<3xi32>, tensor<2xi32>) -> tensor<3x3xi32>
605 %0 = "tf.BroadcastTo"(%input, %shape) : (tensor<3xi16>, tensor<2xi32>) -> tensor<3x3xi16>
615 %0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<3xf32>, tensor<*xi32>) -> tensor<3x3xf32>
625 %0 = "tf.BroadcastTo"(%input, %shape) : (tensor<3xi32>, tensor<2xi32>) -> tensor<*xi32>
636 …%0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<*xi32>) -> tensor<7x8x1x2x3…
640 // CHECK: [[BCT:%.*]] = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<*xi32>) …
645 %0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<8xi32>) -> tensor<*xf32>
649 // CHECK: [[BCT:%.*]] = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<8xi32>) …
654 %0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<*xi32>) -> tensor<*xf32>
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | broadcast_to.h | 62 inline void BroadcastTo(const RuntimeShape& unextended_input_shape, in BroadcastTo() function
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BroadcastTo.pbtxt | 2 graph_op_name: "BroadcastTo"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to.cc | 119 reference_ops::BroadcastTo<kMaxDims>( in Eval()
|