Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for input_type (0.47 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

          return failure();
    
        Value input = tf_op.getInput();
        RankedTensorType input_type =
            mlir::dyn_cast<RankedTensorType>(input.getType());
        // Only rank size four input will be only available by the tf.Conv2D
        // operator verification.
        if (!input_type || input_type.isDynamicDim(3)) {
          return failure();
        }
        // Check if the given op is based on grouped convolution.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

            break;
          default:
            return nullptr;  // Not yet supported
        }
      } else {
        return nullptr;  // Not yet supported
      }
    
      input_type = input_type.clone(new_storage_type);
      return input_type;
    }
    
    // Replaces quant.qcast op to composite quantize_i8 function.
    class ReplaceQuantizePattern
        : public mlir::OpRewritePattern<quantfork::QuantizeCastOp> {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc

    TfLiteStatus QuantizeModel(ModelT* model, const TensorType& input_type,
                               const TensorType& output_type, bool allow_float,
                               std::string& output_buffer) {
      return QuantizeModel(model, input_type, output_type, allow_float,
                           /*operator_names=*/{}, TensorType_INT8, output_buffer);
    }
    
    TfLiteStatus QuantizeModel(ModelT* model, const TensorType& input_type,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

        return in_placeholder, output_tensor
    
      def _create_simple_tf1_gather_model(
          self, input_type: dtypes.DType, use_variable_for_filter=False
      ) -> Tuple[core.Tensor, core.Tensor]:
        """Creates a basic gather model.
    
        This is intended to be used for TF1 (graph mode) tests.
    
        Args:
          input_type: type of the input index tensor for gather operation.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

        // as its an input requirement.
        if (!input_ty.hasRank() || input_ty.getRank() != 4) {
          return failure();
        }
    
        int64_t batch_cst = input_ty.getShape()[0];
        int64_t channels_cst = input_ty.getShape()[3];
    
        int64_t in_y_cst = input_ty.getShape()[1];
        int64_t in_x_cst = input_ty.getShape()[2];
        int64_t in_spatial_cst =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py

          ('use_constant_with_int64_input', np.int64, False),
          ('use_variable_with_int64_input', np.int64, True),
      )
      @test_util.run_v2_only
      def test_gather_model(self, input_type, use_variable):
        model = self._create_gather_model(input_type, use_variable)
    
        save.save(model, self._input_saved_model_path)
    
        rng = np.random.default_rng(seed=42)
        static_input_shape = [6]
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 51.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      if (inserted) {
        NodeDef arg_def;
        NodeDefBuilder builder(
            absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp,
            NodeDebugInfo(src_node->def()));
        DataType dtype = edge->dst()->input_type(edge->dst_input());
        builder.Attr("T", dtype);
        builder.Attr("index", arg_index);
        Status s = builder.Finalize(&arg_def);
        if (!s.ok()) return s;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

    TFR_BUILTINS = {
        '_tfr_quant_act_range': (TFRTypes.TENSOR, TFRTypes.TENSOR),
        '_tfr_quant_rescale': TFRTypes.TENSOR,
        '_tfr_quant_raw_data': lambda input_type: input_type,
        '_tfr_quant_qparam': (TFRTypes.TENSOR, TFRTypes.TENSOR),
        '_tfr_quant_scale_factor': TFRTypes.TENSOR,
    }
    
    
    class TFRTypeResolver(type_inference.Resolver):
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  9. tensorflow/cc/gradients/math_grad.cc

        return errors::InvalidArgument("Cast requires 2 arguments");
      }
      if (grad_inputs.size() != 1) {
        return errors::InvalidArgument("Cast grad requires 1 grad input");
      }
    
      auto src_type = op.input_type(0);
      auto dst_type = grad_inputs[0].type();
      if (IsFloatingPointDtype(src_type) && IsFloatingPointDtype(dst_type)) {
        grad_outputs->push_back(Cast(scope, grad_inputs[0], src_type));
      } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  10. pkg/workloadapi/workload.pb.go

    	11, // 20: istio.workload.Workload.ServicesEntry.value:type_name -> istio.workload.PortList
    	21, // [21:21] is the sub-list for method output_type
    	21, // [21:21] is the sub-list for method input_type
    	21, // [21:21] is the sub-list for extension type_name
    	21, // [21:21] is the sub-list for extension extendee
    	0,  // [0:21] is the sub-list for field type_name
    }
    
    Registered: Fri Jun 14 15:00:06 UTC 2024
    - Last Modified: Wed Jun 12 18:02:35 UTC 2024
    - 65.9K bytes
    - Viewed (0)
Back to top