- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 126 for getFlow (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/init_text_file_to_import.cc
op.getLoc(), DenseStringElementsAttr::get( RankedTensorType::get(static_cast<int64_t>(lines.size()), StringType::get(rewriter.getContext())), lines)); Value value_constant_tensor = rewriter.create<arith::ConstantOp>( op.getLoc(), rewriter.getI64TensorAttr(line_nums));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
return failure(); } auto op = builder.create<CustomOp>( func_.getLoc(), func_.getFunctionType().getResults(), func_.getArguments(), kMaxUnpooling, CustomOption(&builder, custom_option_buffer)); builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); return success(); } LogicalResult ConvertMaxUnpoolingFunc::VerifySignature() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils_test.cc
OpBuilder builder(&module->getBodyRegion()); context.loadDialect<TF::TensorFlowDialect>(); Value value = CreateConstValue<int8_t>(builder, module->getLoc(), unpacked_shape, unpacked_values); Value packed_value = PackOperand(builder, module->getLoc(), value, pack_dim); DenseIntElementsAttr packed_value_attr; ASSERT_TRUE(matchPattern(packed_value, m_Constant(&packed_value_attr)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc
ShapedType shaped_type = RankedTensorType::get({1}, TF::StringType::get(builder.getContext())); auto const_op = builder.create<TF::ConstOp>( asset.getLoc(), DenseStringElementsAttr::get(shaped_type, {filename})); for (auto init_op : init_table_from_text_file_ops_to_erase) { // Replace the InitializeTableFromTextFileV2Op to use the saved model's
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc
mlir::quant::QuantizedType::castToExpressedType(input_type); builder->setInsertionPoint(op); auto dequantize_op = builder->create<TFL::DequantizeOp>( op->getLoc(), dequantized_input_type, input.get()); dequantized_inputs.push_back(dequantize_op); } else { dequantized_inputs.push_back(input.get()); } } // Result types.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
mlir::isa<quant::QuantizedType>(element_type)) { return; } op_builder->setInsertionPoint(const_op); Value scalar = op_builder->create<mhlo::ConstantOp>( const_op->getLoc(), DenseElementsAttr::get( RankedTensorType::get(/*shape=*/{}, element_type), splat_elements_attr.getSplatValue<Attribute>())); auto broadcast_dims = DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
BlockArgument param = callee.getArgument(i); if (mlir::isa<TF::ResourceType>(getElementTypeOrSelf(param.getType()))) { removed_params.set(i); callee.getBody().addArgument(param.getType(), param.getLoc()); param.replaceAllUsesWith(callee.getArguments().back()); removed_params.push_back(false); } } // Remove old resource-type parameters. callee.getBody().front().eraseArguments(removed_params);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/op_or_arg_name_mapper.cc
auto name_from_loc = mlir::GetNameFromLoc(op->getLoc()); if (!name_from_loc.empty()) return name_from_loc; // If the location is none of the expected types, then simply use name // generated using the op type. return std::string(op->getName().getStringRef()); } auto val = op_or_val.dyn_cast<mlir::Value>(); auto name_from_loc = mlir::GetNameFromLoc(val.getLoc()); if (!name_from_loc.empty()) return name_from_loc;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
} static Value CreateTranspose(OpBuilder& builder, Value source, ArrayRef<int64_t> perm) { return builder.create<stablehlo::TransposeOp>(source.getLoc(), source, perm) ->getResult(0); } // Transform pad(transpose(x)) to transpose(pad(x)) struct TransposeCommuteWithPad : public OpRewritePattern<stablehlo::PadOp> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf_while.cc
// Create new TFL While op that will be used to replace TF While op. auto new_op = OpBuilder(op).create<TFL::WhileOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), while_op.getIsStateless()); Location loc = while_op->getLoc(); CreateRegionWithCall(while_op.cond_function(), new_op.getCond(), loc); CreateRegionWithCall(while_op.body_function(), new_op.getBody(), loc);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0)