Searched refs:branch_1 (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/ |
| D | case_op.mlir | 8 // CHECK: func @branch_1
19 …%result = "tf.Case"(%index, %input) {branches = [@branch_0, @branch_1, @branch_2, @branch_3, @bran…
31 func.func @branch_1(%arg0: tensor<i32>) -> tensor<i32> {
|
| /external/libaom/av1/encoder/ |
| D | partition_strategy.c | 242 const float *branch_1 = branch_0 + CNN_BRANCH_0_OUT_SIZE; in av1_intra_mode_cnn_partition() local
243 const float *branch_2 = branch_1 + CNN_BRANCH_1_OUT_SIZE; in av1_intra_mode_cnn_partition()
255 dnn_features[f_idx++] = branch_1[lin_idx + ch_idx * spa_stride]; in av1_intra_mode_cnn_partition()
268 dnn_features[f_idx++] = branch_1[curr_lin_idx + ch_idx * spa_stride]; in av1_intra_mode_cnn_partition()
277 dnn_features[f_idx++] = branch_1[prev_lin_idx + ch_idx * prev_spa_stride]; in av1_intra_mode_cnn_partition()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tensor_list_ops_decomposition.mlir | 259 …%case_op = "tf.Case"(%arg0, %tl) {branches = [@branch_0, @branch_1, @branch_2], is_stateless = fal…
279 // CHECK: func @branch_1(%[[EARG0:.*]]: tensor<10xf32>, %[[EARG1:.*]]: tensor<1xi32>) -> (tensor<10…
280 func.func @branch_1(%arg0: tensor<!tf_type.variant<tensor<f32>>>) -> tensor<!tf_type.variant<tensor…
|
| D | resource_op_lifting.mlir | 425 …%3:2 = "tf.Case"(%arg0, %0, %1) {branches = [@branch_0, @branch_1, @branch_2], is_stateless = fals…
448 // CHECK: func @branch_1(%[[EARG0:.*]]: tensor<4xf32>, %[[EARG1:.*]]: tensor<4xf32>)
449 func.func @branch_1(%arg0: tensor<*x!tf_type.resource<tensor<4xf32>>>, %arg1: tensor<*x!tf_type.res…
|
| D | shape_inference.mlir | 225 …%0 = "tf.Case"(%arg0, %arg1) {branches = [@branch_0, @branch_1], is_stateless = false} : (tensor<i…
237 // CHECK-LABEL: func @branch_1
239 func.func @branch_1(%arg0: tensor<!tf_type.resource>) -> tensor<*xf32> {
|
| /external/vixl/src/aarch64/ |
| D | macro-assembler-aarch64.h | 205 static bool IsValidComparison(const BranchInfo& branch_1, in IsValidComparison() argument
209 if (branch_1.branch_type_ != branch_2.branch_type_) { in IsValidComparison()
216 bool same_offsets = (branch_1.pc_offset_ == branch_2.pc_offset_); in IsValidComparison()
217 return (!same_offsets || ((branch_1.label_ == branch_2.label_) && in IsValidComparison()
218 (branch_1.first_unreacheable_pc_ == in IsValidComparison()
|