Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 52 for call_op (0.12 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

        auto transpose_op = *transpose_ops.begin();
        auto result_type =
            mlir::dyn_cast_or_null<ShapedType>(transpose_op.getResult().getType());
        auto is_valid_move =
            llvm::all_of(op->getOperands(), [result_type](Value operand) -> bool {
              auto operand_type =
                  mlir::dyn_cast_or_null<ShapedType>(operand.getType());
              return result_type && operand_type && result_type.hasRank() &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir

      // CHECK: tf.AssignVariableOp
      %stack = "tf.StackV2"(%max_size) {elem_type = f32, stack_name = "s"} : (tensor<i32>) -> tensor<!tf_type.resource>
      // CHECK: %[[CASE_OUTPUT:.*]] = "tf.CaseRegion"(%[[BRANCH_INDEX]]) {{.*}} ({
      %case_op = "tf.CaseRegion"(%arg0) ({
        %elem = "tf._SomeOp"() : () -> tensor<f32>
        // CHECK-NOT: tf.StackPushV2
        // CHECK: %[[BUFFER_VAL:.*]] = "tf.ReadVariableOp"(%[[BUFFER]])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 25.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/xla_launch_util.cc

        OpKernelContext* ctx) {
      std::vector<int> constant_input_indices;
      TF_RETURN_IF_ERROR(GetCompileTimeConstInputs(
          &ctx->op_kernel(), &constant_input_indices, ctx->function_library()));
      if (!absl::c_all_of(constant_input_indices, [&](int idx) {
            return ctx->input_memory_type(idx) == HOST_MEMORY;
          })) {
        return errors::Internal("Unexpected device placement for a constant input");
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc

          if (!device_list)
            return op.emitError()
                   << "expects 'devices' to be a map alias and device name list.";
    
          bool is_device_string = llvm::all_of(device_list, [](Attribute attr) {
            return mlir::dyn_cast_or_null<StringAttr>(attr);
          });
          if (!is_device_string)
            return op.emitOpError() << "expects 'devices' to be a consists of "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.4K bytes
    - Viewed (0)
  10. src/cmd/vendor/golang.org/x/sys/unix/zsyscall_aix_ppc64_gccgo.go

    func callclose(fd int) (r1 uintptr, e1 Errno) {
    	r1 = uintptr(C.close(C.int(fd)))
    	e1 = syscall.GetErrno()
    	return
    }
    
    // THIS FILE IS GENERATED BY THE COMMAND AT THE TOP; DO NOT EDIT
    
    func calldup(oldfd int) (r1 uintptr, e1 Errno) {
    	r1 = uintptr(C.dup(C.int(oldfd)))
    	e1 = syscall.GetErrno()
    	return
    }
    
    // THIS FILE IS GENERATED BY THE COMMAND AT THE TOP; DO NOT EDIT
    
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Oct 19 23:33:33 UTC 2023
    - 30.9K bytes
    - Viewed (0)
Back to top