- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for constop (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
loc, mlir::RankedTensorType::get({}, getElementTypeOrSelf(shape)), shape, /*begin=*/ builder->create<TF::ConstOp>(loc, builder->getI32TensorAttr({idx})), /*end=*/ builder->create<TF::ConstOp>(loc, builder->getI32TensorAttr({idx + 1})), /*strides=*/ builder->create<TF::ConstOp>(loc, builder->getI32TensorAttr({1})), /*begin_mask=*/0, /*end_mask=*/0, /*ellipsis_mask=*/0,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
state.padding = rewriter.getStringAttr("VALID"); } auto conv_op = static_cast<const ConcreteType *>(this)->createTFLOp( &state, rewriter, op->getLoc(), tf_op.getType(), input, filter, bias); rewriter.replaceOp(op, conv_op.getResult()); return success(); } const IntegerAttr intAttrOne; private: bool allow_bf16_and_f16_type_legalization_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
explicit QuantizeCompositeFunctionsPass() = default; explicit QuantizeCompositeFunctionsPass( const QuantMethod quantization_method, const OpSet target_opset, const bool enable_per_channel_quantization, const int min_num_elements_for_weights, const bool enable_legacy_weight_only, std::optional<const std::string> mlir_dump_file_name) : enable_legacy_weight_only_(enable_legacy_weight_only),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
return CreateI32SplatConst(op.getLoc(), rewriter, {1}, attr.getInt()); } if (auto const_op = num_elements.getDefiningOp<TF::ConstOp>()) { return CreateI32SplatConst(op->getLoc(), rewriter, {1}, (*const_op.getValue() .cast<DenseElementsAttr>() .getValues<APInt>()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
// TODO(krzysd) Handle function calls StatusOr<Operation*> ConvertOp( const tflite::OperatorT& op, const std::vector<Value>& vals_map, const std::vector<mlir::TensorType>& intermediate_types, Value optional_arg_marker, const std::vector<std::unique_ptr<tflite::OperatorCodeT>>& op_codes, const std::vector<std::string>& func_names, const std::vector<std::unique_ptr<tflite::TensorT>>& tensors, Location loc,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
void SetAllVarIsInitializedToTrue(Block* block) { auto builder = OpBuilder::atBlockBegin(block); TF::ConstOp const_true = nullptr; for (auto op : llvm::make_early_inc_range(block->getOps<TF::VarIsInitializedOp>())) { builder.setInsertionPoint(op); if (!const_true) const_true = builder.create<TF::ConstOp>( op.getLoc(), DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
return func_op; } TF::StatefulPartitionedCallOp EncapsulateOpsInFunc( OpBuilder& builder, SymbolTable& symbol_table, const llvm::SetVector<Operation*>& ops, const llvm::SetVector<Value>& inputs, const llvm::SetVector<Value>& outputs, func::FuncOp parent_func, ModuleOp module, const std::string& name, bool flag_for_inlining) { // Moves all of the Operations in 'ops' into a newly created func.FuncOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc
ASSERT_THAT(status, Eq(kTfLiteOk)); const auto& subgraph = model_.subgraphs[0]; auto conv_op = subgraph->operators[0].get(); const int input_tensor_idx = 0; const int weights_tensor_idx = 1; const int bias_tensor_index = 2; const int output_tensor_idx = 0; const auto bias_tensor = subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); const auto input_tensor =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 73.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
return false; } // Returns true if any op in `cluster_ops` has outputs consumed by ops not // `cluster_ops` with a non-static shape. bool HasDynamicOutputs(const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { for (Operation* op : cluster_ops) { for (const OpOperand& use : op->getUses()) { if (cluster_ops.count(use.getOwner())) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions.mlir
%3 = stablehlo.add %2, %1 : tensor<1x3xf32> func.return %3: tensor<1x3xf32> } // CHECK: %[[CONST_0:.*]] = stablehlo.constant dense<2.000000e+00> // CHECK: %[[CONST_1:.*]] = stablehlo.constant dense<2.000000e+00> // CHECK: %[[XLA_CALL_MODULE:.*]] = "tf.XlaCallModule"(%arg0, %[[CONST_0]], %[[CONST_1]]) // CHECK: return %[[XLA_CALL_MODULE:.*]] : tensor<1x3xf32> // CHECK: }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 49.8K bytes - Viewed (0)