- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 57 for call_op (0.25 sec)
-
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
// begin[i]. // // If slice_size[i] < -1 then executing the slice will throw an error, and we // don't do anything here. We've already filtered these cases out in // IsRewritableSlice. if (absl::c_all_of(slice_inputs.size_as_vector, [](int64_t i) { return i >= 0; })) { *size = slice_inputs.size; return absl::OkStatus(); } Output input_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
} for (auto pair : llvm::zip(new_op.getRegions(), case_op.getRegions())) { std::get<0>(pair)->takeBody(*std::get<1>(pair)); } case_op.replaceAllUsesWith( new_op.getResults().take_front(case_op.getNumResults())); case_op.erase(); return success(); } template <typename CallOp> LogicalResult HandlePartitionedCallOp( CallOp call, func::FuncOp callee, ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
src/cmd/compile/internal/typecheck/stmt.go
// and spilled to temporary variables, which can be captured by // the wrapper function. stmtPos := base.Pos callPos := base.Pos as := ir.NewAssignListStmt(callPos, ir.OAS2, make([]ir.Node, len(argps)), make([]ir.Node, len(argps))) for i, argp := range argps { arg := *argp pos := callPos if ir.HasUniquePos(arg) { pos = arg.Pos() } // tmp := arg
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Mon Nov 20 15:10:54 UTC 2023 - 17.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
TF::UniformQuantizedClipByValueOp>(); target.addDynamicallyLegalOp<TF::CastOp>([](Operation *op) { auto cast_op = llvm::dyn_cast<TF::CastOp>(op); return !IsTFQintType(cast_op.getSrcT()) && !IsTFQintType(cast_op.getDstT()); }); RewritePatternSet patterns(ctx); PopulateLegalizeTfQuantizationPatterns(ctx, &patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc
filename_tensor_name, restore_op_name, save_node_name}; const auto is_empty_predicate = [](const absl::string_view s) { return s.empty(); }; if (absl::c_all_of(fields, is_empty_predicate)) { return std::nullopt; } else if (absl::c_none_of(fields, is_empty_predicate)) { SaverDef saver_def{}; saver_def.set_version(SaverDef::V2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
int padding_after = padding_before + offset; padding_values[2 * i] = padding_before; padding_values[2 * i + 1] = padding_after; } } if (input_zp_value == 0 || absl::c_all_of(padding_values, [](int v) { return v == 0; })) { padding = CreateConstValue<int32_t>( builder, loc, {num_dims - 2, 2}, SmallVector<int32_t>(padding_values.begin() + 2,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
return uniform_type.getZeroPoint() == 0; } else if (auto per_axis_type = llvm::dyn_cast_or_null<UniformQuantizedPerAxisType>(qtype)) { return absl::c_all_of(per_axis_type.getZeroPoints(), [](int64_t x) { return x == 0; }); } return false; } // Multiplies two 1D arrays with broadcasting support. template <typename T>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass.cc
for (Node* n : rpo) { if (!compile_time_const_nodes[n->id()]) { continue; } absl::string_view cluster_name = *GetXlaClusterForNode(*n); bool node_on_cluster_edge = absl::c_all_of(n->in_edges(), [&](const Edge* e) { std::optional<absl::string_view> incoming_cluster = GetXlaClusterForNode(*e->src()); return !incoming_cluster || *incoming_cluster != cluster_name;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/report_test.cc
EXPECT_TRUE(result.method().has_static_range_ptq()); } TEST_F(QuantizationReportTest, InitializeWithModuleOpWithoutQuantizationMethodAttribute) { // A quantized dot_general op but the `CallOp` is missing the // `_quantization_method` attribute. constexpr absl::string_view kQuantizedDotGeneralMissingQuantizationMethodAttr = R"mlir( func.func @main(%arg0: tensor<1x2xf32>) -> tensor<1x3xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 10:10:34 UTC 2024 - 18.5K bytes - Viewed (0)