| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | inception_resnet_v2.py | 155 branch_1 = conv2d_bn(x, 48, 1)
156 branch_1 = conv2d_bn(branch_1, 64, 5)
162 branches = [branch_0, branch_1, branch_2, branch_pool]
173 branch_1 = conv2d_bn(x, 256, 1)
174 branch_1 = conv2d_bn(branch_1, 256, 3)
175 branch_1 = conv2d_bn(branch_1, 384, 3, strides=2, padding='valid')
177 branches = [branch_0, branch_1, branch_pool]
188 branch_1 = conv2d_bn(x, 256, 1)
189 branch_1 = conv2d_bn(branch_1, 288, 3, strides=2, padding='valid')
194 branches = [branch_0, branch_1, branch_2, branch_pool]
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | choose_fastest_branch_dataset_test.py | 62 def branch_1(dataset): function
66 dataset, [branch_0, branch_1])
78 def branch_1(dataset): function
82 dataset, [branch_0, branch_1], ratio_numerator=10)
95 def branch_1(dataset): function
99 dataset, [branch_0, branch_1], ratio_numerator=10)
113 def branch_1(dataset): function
117 dataset, [branch_0, branch_1])
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | choose_fastest_branch_dataset_serialization_test.py | 46 def branch_1(dataset): function
50 dataset, [branch_0, branch_1],
67 def branch_1(dataset): function
71 dataset, [branch_0, branch_1], num_elements_per_branch=3)
86 def branch_1(dataset): function
90 dataset, [branch_0, branch_1], num_elements_per_branch=3)
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | choose_fastest_branch_benchmark.py | 32 branch_1, argument
36 ds_1 = branch_1(input_dataset)
38 input_dataset, [branch_0, branch_1],
51 def branch_1(dataset): function
54 return self.make_benchmark_datasets(dataset, branch_0, branch_1, 100)
|
| /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 @branch_1(%arg0: tensor<i32>) -> tensor<i32> {
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 267 const FunctionDef* branch_1 = in CheckVectorizedWithChooseFastest() local
269 ASSERT_NE(branch_1, nullptr); in CheckVectorizedWithChooseFastest()
270 CheckBranch(*branch_1, expected_original_branch); in CheckVectorizedWithChooseFastest()
454 const FunctionDef* branch_1 = in TEST_P() local
456 ASSERT_NE(branch_1, nullptr); in TEST_P()
457 CheckBranch(*branch_1, original_ops); in TEST_P()
|
| /external/libaom/libaom/av1/encoder/ |
| D | partition_strategy.c | 152 const float *branch_1 = branch_0 + CNN_BRANCH_0_OUT_SIZE; in av1_intra_mode_cnn_partition() local
153 const float *branch_2 = branch_1 + CNN_BRANCH_1_OUT_SIZE; in av1_intra_mode_cnn_partition()
165 dnn_features[f_idx++] = branch_1[lin_idx + ch_idx * spa_stride]; in av1_intra_mode_cnn_partition()
178 dnn_features[f_idx++] = branch_1[curr_lin_idx + ch_idx * spa_stride]; in av1_intra_mode_cnn_partition()
187 dnn_features[f_idx++] = branch_1[prev_lin_idx + ch_idx * prev_spa_stride]; in av1_intra_mode_cnn_partition()
|
| /external/llvm-project/mlir/test/Rewrite/ |
| D | pdl-bytecode.mlir | 138 // CHECK-LABEL: test.branch_1
140 module @ir attributes { test.branch_1 } {
|
| /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 @branch_1(%arg0: tensor<!tf.variant<tensor<f32>>>) -> tensor<!tf.variant<tensor<f32>>> {
|
| 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 @branch_1(%arg0: tensor<!tf.resource>) -> tensor<*xf32> {
|
| D | resource_op_lifting.mlir | 400 …%3:2 = "tf.Case"(%arg0, %0, %1) {branches = [@branch_0, @branch_1, @branch_2], is_stateless = fals…
423 // CHECK: func @branch_1(%[[EARG0:.*]]: tensor<4xf32>, %[[EARG1:.*]]: tensor<4xf32>)
424 func @branch_1(%arg0: tensor<*x!tf.resource<tensor<4xf32>>>, %arg1: tensor<*x!tf.resource<tensor<4x…
|
| /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()
|