Searched refs:cond_type (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | select_v2_test_util.cc | 32 template <DataType cond_type>
33 std::vector<float> SetUpData(Tensor<BHWC, cond_type>& cond_tensor, in SetUpData()
106 template <DataType cond_type>
111 const Tensor<BHWC, cond_type>& cond_tensor, in RunSelectV2()
118 op_def.src_tensors.push_back({cond_type, storage, layout}); in RunSelectV2()
140 template <DataType cond_type>
147 Tensor<BHWC, cond_type> cond_tensor; in SelectV2TestTemplate()
150 std::vector<float> expected_data = SetUpData<cond_type>( in SelectV2TestTemplate()
159 RETURN_IF_ERROR(RunSelectV2<cond_type>( in SelectV2TestTemplate()
169 template <DataType cond_type>
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_functional_ops.cc | 113 auto cond_type = cond.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
114 if (!cond_type || !cond_type.getShape().equals({}) || in matchAndRewrite()
115 !cond_type.getElementType().isInteger(/*width=*/1)) in matchAndRewrite()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | loop_unswitch_pass.cpp | 299 const analysis::Type* cond_type = in PerformUnswitch() local
309 cst_mgr->GetDefiningInstruction(cst_mgr->GetConstant(cond_type, {0})), in PerformUnswitch()
312 cst_mgr->GetDefiningInstruction(cst_mgr->GetConstant(cond_type, {1})); in PerformUnswitch()
322 cond_type, iv_condition->GetInOperand(i).words)), in PerformUnswitch()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/ |
| D | loop_unswitch_pass.cpp | 292 const analysis::Type* cond_type = in PerformUnswitch() local
302 cst_mgr->GetDefiningInstruction(cst_mgr->GetConstant(cond_type, {0})), in PerformUnswitch()
305 cst_mgr->GetDefiningInstruction(cst_mgr->GetConstant(cond_type, {1})); in PerformUnswitch()
315 cond_type, iv_condition->GetInOperand(i).words)), in PerformUnswitch()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | loop_unswitch_pass.cpp | 294 const analysis::Type* cond_type = in PerformUnswitch() local
304 cst_mgr->GetDefiningInstruction(cst_mgr->GetConstant(cond_type, {0})), in PerformUnswitch()
307 cst_mgr->GetDefiningInstruction(cst_mgr->GetConstant(cond_type, {1})); in PerformUnswitch()
317 cond_type, iv_condition->GetInOperand(i).words)), in PerformUnswitch()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3150 auto cond_type = in verify() local
3152 if (!cond_type || !cond_type.getShape().equals({}) || in verify()
3153 !cond_type.getElementType().isInteger(/*width=*/1)) in verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2954 auto cond_type = op.condition().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
2957 if (!cond_type || !then_type || !else_type) { in matchAndRewrite()
2971 bool needs_broadcast = cond_type.getRank() == 1 && then_type.getRank() != 1; in matchAndRewrite()
2985 if (cond_type.getRank() > 0) { in matchAndRewrite()
3006 result_extents, GetI64ElementsAttrForSeq(0, cond_type.getRank(), &b)); in matchAndRewrite()
|