- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 183 for setLoc (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
auto new_buffer = builder.create<TF::IdentityOp>( pop.getLoc(), ArrayRef<Type>{buffer.getType()}, ArrayRef<Value>{buffer}); auto new_size = builder.create<TF::SubOp>( pop.getLoc(), ArrayRef<Type>{size.getType()}, ArrayRef<Value>{size, cutil::GetR1Const({1LL}, builder, pop.getLoc())}); auto element = cutil::GetElement(new_size, new_buffer, builder, pop.getLoc()); pop.getOutputHandle().replaceAllUsesWith(new_buffer);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
auto dcast = b.create<quantfork::DequantizeCastOp>( dq.getLoc(), dq.getOutput().getType(), dq.getInput()); dq.getOutput().replaceAllUsesWith(dcast); dq.erase(); } else if (auto q = llvm::dyn_cast<QuantizeOp>(op)) { auto qcast = b.create<quantfork::QuantizeCastOp>( q.getLoc(), q.getOutput().getType(), q.getInput()); q.getOutput().replaceAllUsesWith(qcast);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// Calculates the broadcast shape using BroadcastArgs op. Value cond_shape = GetShape(cond, op->getLoc(), rewriter); Value lhs_shape = GetShape(lhs, op->getLoc(), rewriter); Value rhs_shape = GetShape(rhs, op->getLoc(), rewriter); auto broadcast_shape_value = rewriter .create<TF::BroadcastArgsOp>(op->getLoc(), lhs_shape.getType(), lhs_shape, rhs_shape)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc
auto new_read_variable_op = builder.create<ReadVariableOp>(read_variable_op.getLoc(), ref_qtype, read_variable_op.getResourceId()); auto new_dq_op = builder.create<DequantizeOp>( read_variable_op.getLoc(), read_variable_op.getResult().getType(), new_read_variable_op.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
cutil::ReadLocalVariable(pop.getHandle(), builder, pop.getLoc()); auto size = cutil::ReadLocalVariable(it->getSecond(), builder, pop.getLoc()); auto new_size = builder.create<TF::SubOp>( pop.getLoc(), ArrayRef<Type>{size.getType()}, ArrayRef<Value>{size, cutil::GetR1Const({1}, builder, pop.getLoc())}); auto pop_val = cutil::GetElement(new_size, stack_val, builder, pop.getLoc()); pop.replaceAllUsesWith(pop_val);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
req_op.getLoc(), func_op.getResultTypes()[0], req_op.getOperand()); if (func_name.contains("_relu6_")) { auto min = rewriter.create<mlir::stablehlo::ConstantOp>( req_op.getLoc(), rewriter.getF32FloatAttr(0)); auto max = rewriter.create<mlir::stablehlo::ConstantOp>( req_op.getLoc(), rewriter.getF32FloatAttr(6));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
auto slice_begin_const = rewriter->create<TFL::ConstOp>(split_op->getLoc(), slice_begin_attr); auto slice_size_const = rewriter->create<TFL::ConstOp>(split_op->getLoc(), slice_size_attr); auto slice_op = rewriter->create<TFL::SliceOp>( split_op->getLoc(), current_output_type, input, slice_begin_const, slice_size_const); // Rewire output.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
auto erf_op = rewriter.create<chlo::ErfOp>(mul_op.getLoc(), mul_op); auto add_op = rewriter.create<stablehlo::AddOp>(erf_op.getLoc(), erf_op, one_cst); auto lhs_mul_op = rewriter.create<stablehlo::MulOp>( half_cst.getLoc(), new_func.getArguments()[0], half_cst); auto output_mul_op = rewriter.create<stablehlo::MulOp>(lhs_mul_op.getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
Value start_op = rewriter.create<TF::ConstOp>(iota_op.getLoc(), start); Value limit_op = rewriter.create<TF::ConstOp>(iota_op.getLoc(), limit); Value delta_op = rewriter.create<TF::ConstOp>(iota_op.getLoc(), delta); Value result = rewriter.create<TF::RangeOp>(iota_op.getLoc(), range_type, start_op, limit_op, delta_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
rewriter.create<func::FuncOp>(input_op->getLoc(), func_name, func_type); OpBuilder::InsertionGuard guard = OpBuilder::InsertionGuard(rewriter); ArrayRef<Type> inputs = quantization_func.getFunctionType().getInputs(); Block* block = rewriter.createBlock( &quantization_func.getBody(), quantization_func.begin(), inputs, SmallVector<Location>(inputs.size(), quantization_func.getLoc())); func_input_arg = block->getArgument(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0)