- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for input_type (0.83 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
callers.backward->getParentOfType<func::FuncOp>(); const std::vector<Value>& operands = loop_operands_nm0; // Input types will be the same as the original loop body. std::vector<Type> input_types = GetValueTypes(operands); // Determine the results types. // Return ALL outputs, respecting the provided order of the Operations. This // makes it straightforward for users of this function to map the return
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/flatbuffer_import.cc
control_nodes.try_emplace(from); control_nodes[to].incoming.insert(from); } llvm::SmallVector<mlir::Type, 2> ret_types; llvm::SmallVector<mlir::Type, 4> input_types; auto func_loc = mlir::NameLoc::get(builder.getStringAttr(name), base_loc); std::vector<int> func_inputs = subgraph.inputs; if (is_entry_point && !ordered_input_arrays.empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)