- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 100 for input_dtype (0.15 sec)
-
tensorflow/c/kernels_experimental.cc
tensorflow::Var** maybe_resource) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); *maybe_resource = nullptr; if (cc_ctx->input_dtype(input) == tensorflow::DT_RESOURCE) { if (LookupResource(cc_ctx, HandleFromInput(cc_ctx, input), maybe_resource) .ok()) { return (*maybe_resource)->mu(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
absl::StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, Builder* builder) { const auto& input_dtype = input_tensor.dtype(); const auto& input_shape = input_tensor.shape(); Type elt_type; TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, *builder, &elt_type)); SmallVector<int64_t, 4> shape; ConvertToMlirShape(input_shape, &shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
tensorflow/c/kernels.cc
auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); CHECK_GE(index, 0); // Crash OK CHECK_LT(index, cc_ctx->num_inputs()); // Crash OK return static_cast<TF_DataType>(cc_ctx->input_dtype(index)); } void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor, TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.h
// process. struct ExpressedToQuantizedConverter { // Creates a converter for the given input type. static ExpressedToQuantizedConverter forInputType(Type input_type); // Converts the inputType to be based on the given elemental type, // returning the new type (or nullptr and emit an error on failure). Type convert(quant::QuantizedType elemental_type) const; // Whether the conversion is legal.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
mlir::OpBuilder builder(func.getBody()); auto& bb = func.front(); auto loc = func.getLoc(); int num_args = bb.getNumArguments(); llvm::SmallVector<Type, 4> input_types; input_types.reserve(num_args); // Edit the block arguments and create the new input ops in place to replace // the old input ops and quantize ops. for (int i = 0; i != num_args; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); SmallVector<mlir::Type, 4> input_types{input_type, weight_type, bias_type, projection_type, layer_norm_scale_type}; auto func_type = builder->getFunctionType(input_types, output_type); auto func = func::FuncOp::create(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
Operation** result) { auto input_type = mlir::cast<RankedTensorType>(input.getType()); SmallVector<int64_t, 4> output_shape; int size_of_splits; if (input_type.getRank() < axis || axis < 0) return failure(); for (int i = 0; i < input_type.getRank(); ++i) { int64_t dim = input_type.getDimSize(i); if (i == axis) { if (dim % splits != 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.cc
return ExpressedToQuantizedConverter{input_type, element_type}; } // Supported primitive type (which just is the expressed type). if (isQuantizablePrimitiveType(input_type)) return ExpressedToQuantizedConverter{input_type, input_type}; // Unsupported. return ExpressedToQuantizedConverter{input_type, nullptr}; } Type ExpressedToQuantizedConverter::convert(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
auto input = splitv_op.getValue(); auto input_type = mlir::dyn_cast_or_null<RankedTensorType>(input.getType()); if (!input_type || !input_type.hasRank()) return failure(); for (auto result : splitv_op.getResults()) { auto result_type = mlir::dyn_cast<RankedTensorType>(result.getType()); if (result_type == nullptr) return failure(); } const int64_t rank = input_type.getRank();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.cc
if (input_op && input_op == from_graph.getOperation()) { auto input_type = mlir::dyn_cast_or_null<RankedTensorType>(input.getType()); if (input_type == nullptr || !input_type.hasStaticShape()) continue; // Quantized type does not support getSizeInBits. if (IsQUI8Type(input_type) || IsQI8Type(input_type)) { total_size_transferred += input_type.getNumElements() * 8; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0)