Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 57 for call_op (0.25 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir

      %case_op = "tf.Case"(%arg0, %tl) {branches = [@branch_0, @branch_1, @branch_2], is_stateless = false}
        : (tensor<i32>, tensor<!tf_type.variant<tensor<f32>>>) -> tensor<!tf_type.variant<tensor<f32>>>
      // CHECK: "tf.Slice"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 38.6K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top