| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | convert_tfl_uint8.cc | 109 0, true /* signed */, builder.getBoolAttr(narrow_range)))); in matchAndRewrite()
183 0, true /* signed */, builder.getBoolAttr(narrow_range))); in convert_graph_uint8_tensor()
197 builder.getBoolAttr(true), builder.getBoolAttr(false), in convert_graph_uint8_tensor()
198 builder.getBoolAttr(false)); in convert_graph_uint8_tensor()
252 0, true /* signed */, builder.getBoolAttr(narrow_range))); in convert_graph_uint8_tensor()
320 builder.getBoolAttr(true), builder.getBoolAttr(false), in convert_graph_uint8_tensor()
321 builder.getBoolAttr(false)); in convert_graph_uint8_tensor()
|
| D | legalize_utils.cc | 43 rewriter.getBoolAttr(true), rewriter.getBoolAttr(double_round), in buildRescale()
44 rewriter.getBoolAttr(false)); in buildRescale()
111 rewriter.getI32ArrayAttr({shift}), rewriter.getBoolAttr(true), in buildRescaleOpConvOutput()
112 rewriter.getBoolAttr(true), rewriter.getBoolAttr(false)); in buildRescaleOpConvOutput()
152 rewriter.getI32ArrayAttr(shift_arr), rewriter.getBoolAttr(true), in buildRescaleOpConvOutput()
153 rewriter.getBoolAttr(true), rewriter.getBoolAttr(true)); in buildRescaleOpConvOutput()
|
| /external/llvm-project/mlir/test/lib/Dialect/Tosa/ |
| D | TosaTestPasses.cpp | 83 rewriter.getBoolAttr(narrow_range))); in matchAndRewrite()
171 rewriter.getBoolAttr(true), rewriter.getBoolAttr(true), in matchAndRewrite()
172 rewriter.getBoolAttr(false)); in matchAndRewrite()
|
| /external/llvm-project/mlir/unittests/Pass/ |
| D | PassManagerTest.cpp | 42 op.setAttr("isFunc", builder.getBoolAttr(ga.isFunc)); in runOnOperation()
43 op.setAttr("isSecret", builder.getBoolAttr(sa.isSecret)); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_attr.cc | 41 return builder->getBoolAttr(value.b()); in ConvertNonFuncAttributeValue()
60 attrs.push_back(builder->getBoolAttr(item)); in ConvertNonFuncAttributeValue()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 68 /*Truncate=*/builder->getBoolAttr(false)); in CallFn()
92 /*Truncate=*/builder->getBoolAttr(false)); in PrepareValsForJump()
125 /*Truncate=*/builder->getBoolAttr(false)); in ReplaceOpResultWithBlockArgs()
|
| D | tf_data_optimization.cc | 42 batchInputOp.num_parallel_calls(), rewriter.getBoolAttr(false)); in matchAndRewrite()
|
| D | unroll_batch_matmul.cc | 171 auto false_attr = rewriter.getBoolAttr(false); in createMatMulOps()
252 auto false_attr = rewriter.getBoolAttr(false); in matchAndRewrite()
|
| D | functional_control_flow_to_regions.cc | 76 /*Truncate=*/builder.getBoolAttr(false)); in CreateCall()
|
| D | einsum.cc | 337 op.getLoc(), matmul_type, lhs, rhs, rewriter.getBoolAttr(false), in rewriteToBatchMatmul()
338 rewriter.getBoolAttr(false)); in rewriteToBatchMatmul()
|
| D | tpu_dynamic_layout_pass.cc | 153 builder->getNamedAttr("is_output", builder->getBoolAttr(false))}); in BuildGetLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 150 /*has_side_effect=*/builder_.getBoolAttr(has_side_effect), in CustomCallInternal()
253 builder_.getI64IntegerAttr(dimension), builder_.getBoolAttr(is_stable)); in SortInternal()
310 builder_.getBoolAttr(indices_are_sorted), in ScatterInternal()
311 builder_.getBoolAttr(unique_indices)); in ScatterInternal()
449 builder_.getBoolAttr(options.left_side()), in TriangularSolveInternal()
450 builder_.getBoolAttr(options.lower()), in TriangularSolveInternal()
451 builder_.getBoolAttr(options.unit_diagonal()), in TriangularSolveInternal()
463 loc_, result_ty, GetValue(a), builder_.getBoolAttr(lower)); in CholeskyInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 325 builder_->getBoolAttr(custom_call->custom_call_has_side_effect()))); in ImportInstructionImpl()
343 builder_->getBoolAttr(instruction->cholesky_options().lower()))); in ImportInstructionImpl()
360 builder_->getBoolAttr(gather_instruction->indices_are_sorted()))); in ImportInstructionImpl()
438 builder_->getBoolAttr(scatter->indices_are_sorted()))); in ImportInstructionImpl()
440 "unique_indices", builder_->getBoolAttr(scatter->unique_indices()))); in ImportInstructionImpl()
497 builder_->getBoolAttr(sort_instruction->is_stable())); in ImportInstructionImpl()
641 builder_->getBoolAttr( in ImportInstructionImpl()
644 "lower", builder_->getBoolAttr( in ImportInstructionImpl()
648 builder_->getBoolAttr( in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 421 sort.is_stableAttr(builder_.getBoolAttr(sort_instr->is_stable())); in EmitSortOp()
570 builder_.getBoolAttr(scatter_instr->indices_are_sorted())); in EmitScatterOp()
572 builder_.getBoolAttr(scatter_instr->unique_indices())); in EmitScatterOp()
656 cholesky_op.is_lowerAttr(builder_.getBoolAttr(options.lower())); in EmitCholesky()
782 builder_.getBoolAttr(backend_config.tensor_ops_enabled()), in EmitDnnConvolution()
1041 op.constrain_layoutAttr(builder.getBoolAttr(collective->constrain_layout())); in SetupCommonCollectiveOpAttributes()
1069 builder_.getBoolAttr(all_gather->use_global_device_ids())); in EmitAllGatherOp()
1083 builder_.getBoolAttr(all_reduce->use_global_device_ids())); in EmitAllReduceOp()
1305 triangular_solve.left_sideAttr(builder_.getBoolAttr(options.left_side())); in EmitTriangularSolveOp()
1306 triangular_solve.lowerAttr(builder_.getBoolAttr(options.lower())); in EmitTriangularSolveOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 127 rewriter.getBoolAttr(false)); in CreateCastToInt32()
282 rewriter.getBoolAttr(false)); in matchAndRewrite()
508 rewriter.getNamedAttr("time_major", rewriter.getBoolAttr(true))); in matchAndRewrite()
566 rewriter.getNamedAttr("time_major", rewriter.getBoolAttr(true))); in matchAndRewrite()
|
| D | optimize.cc | 396 /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.keep_num_dims())); in matchAndRewrite()
420 rewriter.getBoolAttr(fully_connected_op.keep_num_dims()); in matchAndRewrite()
504 /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.keep_num_dims())); in matchAndRewrite()
|
| D | prepare_quantize.cc | 186 BoolAttr narrow_range = builder.getBoolAttr(false); in SetInputNodesQuantizationParams()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/ |
| D | tf_to_quant.cc | 111 BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.narrow_range()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tensorflow_abi_knowledge_propagation.cc | 144 b.getBoolAttr(true)); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| D | Utils.cpp | 95 ifOp.setAttr(idForIfOp, b.getBoolAttr(true)); in hoistAffineIfOp()
|
| /external/llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| D | ShapeToStandard.cpp | 259 rewriter.create<ConstantOp>(loc, i1Ty, rewriter.getBoolAttr(true)); in matchAndRewrite()
462 Value init = b.create<ConstantOp>(loc, i1Ty, b.getBoolAttr(true)); in matchAndRewrite()
479 Value result = b.create<ConstantOp>(loc, i1Ty, b.getBoolAttr(false)); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 90 BoolAttr Builder::getBoolAttr(bool value) { in getBoolAttr() function in Builder
223 values, [this](bool v) -> Attribute { return getBoolAttr(v); })); in getBoolArrayAttr()
|
| /external/llvm-project/mlir/unittests/TableGen/ |
| D | OpBuildGen.cpp | 42 builder.getBoolAttr(true)), in OpBuildGenTest()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.cc | 186 return builder.getBoolAttr(value); in BuildBoolAttr()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertLaunchFuncToLLVMCalls.cpp | 65 loc, llvmI1Type, builder.getBoolAttr(false)); in copy()
|