| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | deduplicate_batch_function.cc | 70 if (!std::equal(lhs_op.getOperandTypes().begin(), in AreEquivalent()
71 lhs_op.getOperandTypes().end(), in AreEquivalent()
72 rhs_op.getOperandTypes().begin())) in AreEquivalent()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/transforms/ |
| D | map_lmhlo_to_scalar_op.h | 34 op.getLoc(), resultTypes, llvm::to_vector<4>(op->getOperandTypes()),
46 llvm::to_vector<4>(op->getOperandTypes()), args, b); in map()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_traits.h | 61 for (auto operand_type : op->getOperandTypes()) { in verifyTrait()
94 for (const auto& operand_type : op->getOperandTypes()) { in verifySameOperandsAndResultElementTypeResolveRef()
|
| D | tf_device.cc | 155 for (auto result_type : region_terminator->getOperandTypes()) { in verify()
375 for (const auto& type : terminator.getOperandTypes()) in parse()
557 llvm::enumerate(terminator.getOperandTypes())) { in verify()
752 op->getLoc(), new_return.getOperandTypes(), op->getOperands(), in EliminatePassThroughResults()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | ensure_static_shapes_pass.cc | 39 llvm::any_of(op->getOperandTypes(), is_dynamic_shape_type)) { in runOnFunction()
|
| D | decompose_reduce_dataset.cc | 130 auto while_input_types = dataset_while.getOperandTypes(); in PopulateDatasetWhileCond()
220 auto while_input_types = dataset_while.getOperandTypes(); in PopulateDatasetWhileBody()
|
| D | shape_inference.cc | 951 for (auto result : zip(op.getResults(), then_yield->getOperandTypes(), in InferShapeForIfRegion()
952 else_yield->getOperandTypes())) { in InferShapeForIfRegion()
977 host_compute_op.getOperandTypes(), in InferShapeForXlaHostComputeMlir()
981 if (failed(PropagateShapeToRegions(host_compute_op.getOperandTypes(), in InferShapeForXlaHostComputeMlir()
1103 for (auto t : llvm::drop_begin(op->getOperandTypes())) { in InferShapeForDatasetOpCommon()
1220 for (auto t : llvm::drop_begin(op.getOperandTypes(), 1 + num_states)) { in InferShapeForReduceDataset()
2087 while_region.body().front().getTerminator()->getOperandTypes()); in InferShapeForSingleOperation()
2197 std::equal(caller->getOperandTypes().begin(), in PropagateShapeToFunctions()
2198 caller->getOperandTypes().end(), in PropagateShapeToFunctions()
2199 callers.front()->getOperandTypes().begin()); in PropagateShapeToFunctions()
[all …]
|
| D | parallel_execute_to_islands.cc | 112 island_op.getLoc(), yield.getOperandTypes(), in ExpandParallelExecuteToIslands()
|
| D | constant_fold.cc | 67 int64_t operands_size = get_size(inst->getOperandTypes()); in ShouldBeFolded()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | bufferize.cc | 61 return converter->isLegal(op->getOperandTypes()) && in populateExtraBufferizePatterns()
|
| D | convert_to_signless_pass.cc | 81 return converter.isLegal(op->getOperandTypes()) && in runOnOperation()
|
| D | tf_to_jit_invocations.cc | 93 op->getOperandTypes(), locs); in matchAndRewrite()
216 yield_op->getOperandTypes())); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | lmhlo_to_jitrt.cc | 201 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
265 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
303 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
423 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
495 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
581 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
995 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
1043 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
1312 FunctionType::get(ctx, op.getOperandTypes(), TypeRange()); in matchAndRewrite()
1512 FunctionType::get(ctx, op->getOperandTypes(), TypeRange()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | insert_main_function.cc | 121 result_types.append(return_op.getOperandTypes().begin(), in CreateMainFunction()
122 return_op.getOperandTypes().end()); in CreateMainFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | rewrite_quantized_io.cc | 104 terminator->getOperandTypes()); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/opdefs/ |
| D | tf_jitrt_ops.cc | 138 for (Type type : op.getOperandTypes()) { in cost()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ir/ |
| D | QuantOps.cc | 81 for (auto input : llvm::zip(getOperandTypes(), getInputSpecs())) { in verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | raise_custom_ops.cc | 90 new_block->addArguments(op->getOperandTypes(), in runOnOperation()
|
| D | optimize_functional_ops.cc | 56 auto return_types = llvm::to_vector<4>(terminator->getOperandTypes()); in UpdateFuncType()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | trivial_spmd_expander.cc | 38 auto output_types = llvm::to_vector<8>(terminator_op.getOperandTypes()); in ExpandOp()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | convert_to_signless_pass.cc | 106 return converter.isLegal(op->getOperandTypes()) && in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | strip_quant_types.cc | 143 for (Type type : op->getOperandTypes()) { in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_clustering.cc | 96 return IsSupportedDataTypes(op->getOperandTypes()); in IsSupportedOperandTypes()
882 bool has_i1_integers = llvm::any_of(op->getOperandTypes(), IsI1Integer) || in VerifyCluster()
890 llvm::any_of(op->getOperandTypes(), IsUnsignedInteger) || in VerifyCluster()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | outline_with_xla_framework.cc | 80 if (llvm::any_of(op->getOperandTypes(), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.h | 80 auto allMatch = llvm::all_of(op->getOperandTypes(), typeMatch) && in verifyTrait()
|