- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 90 for input_type (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
// Do not create any new function for the operations on the localhost. if (IsOnLocalHost(host)) continue; llvm::SmallVector<mlir::Type, 4> input_types; llvm::SmallVector<mlir::Type, 4> result_types; for (Value input : metadata.inputs) { input_types.push_back(input.getType()); } for (Value result : metadata.results) { result_types.push_back(result.getType()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
// within the subgraph and are referenced outside the subgraph. llvm::SmallVector<Type> input_types = TypesFromValues(subgraph.FuncArguments()); llvm::SmallVector<Type> return_types = TypesFromValues(subgraph.FuncOutputs()); FunctionType function_type = builder.getFunctionType(input_types, return_types); std::string function_name = absl::StrCat("func_", subgraph.subgraph_id_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
auto* context = shape_refiner_->GetContext(&node); DataType dtype = node.input_type(idx); return ConvertDataTypeAndShape(dtype, context->input(idx), context->input_handle_shapes_and_types(idx), context, builder); } DataType dtype = node.properties()->input_types[idx]; mlir::Type element_type;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
mlir::MLIRContext* context = tfr_module_.getContext(); llvm::SmallVector<mlir::Type, 4> input_tys, output_tys; mlir::Builder builder(context); for (auto ty : input_dtys) { mlir::Type elt_ty; TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); mlir::TensorType mlir_ty = mlir::UnrankedTensorType::get(elt_ty); input_tys.push_back(mlir_ty); } for (auto ty : output_dtys) { mlir::Type elt_ty;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
// attaching a to the initializer function's type so that it is identifiable. Location CreateInitOpLoc(MLIRContext* ctx, func::FuncOp init_func_ops) { const std::string init_type = GetInitializerType(init_func_ops); const std::string name = absl::StrCat(init_type, "_", init_func_ops.getName().str()); return NameLoc::get(StringAttr::get(ctx, name)); } void MergeInitializerFunctionOpsToMainPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
enable_shape_inference, unconditionally_use_set_output_shapes, enable_soft_placement, set_original_tf_func_name}; auto module_or = tensorflow::GraphdefToMlirTranslateFunction( input, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, options, context); if (!module_or.status().ok()) return nullptr; return std::move(module_or).value(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K 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/compiler/mlir/lite/tf_to_tfl_flatbuffer.h
bool use_splatted_constant, const std::vector<std::string>& extra_tf_opdefs, const GraphImportConfig& specs, absl::string_view debug_info_file, absl::string_view input_arrays, absl::string_view input_dtypes, absl::string_view input_shapes, absl::string_view output_arrays, absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr, mlir::MLIRContext* context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 08:30:24 UTC 2024 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
return GraphdefToSplattedMlirTranslateFunction( file->getBuffer(), input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, graphdef_conversion_options, context); } return GraphdefToMlirTranslateFunction(file->getBuffer(), input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
// Build the function type. Region arguments and extern values together // become the function arguments, with region arguments going first. auto input_types = llvm::to_vector<4>(entry.getArgumentTypes()); for (auto input : extern_values) input_types.push_back(input.getType()); // Terminator operands and pass through extern values (if enabled) together // become the function return values.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0)