- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 81 for _input_shapes (0.62 sec)
-
tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc
return func; } func::FuncOp createMaxUnpoolingFunc( mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape, const SmallVector<int64_t, 4>& output_shape) { auto input_type = RankedTensorType::get(input_shape, builder->getF32Type()); auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); auto output_type = RankedTensorType::get(output_shape, builder->getF32Type());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
arg_kinds_vector)) { XlaArgument& arg = std::get<0>(arg_components); TensorShape shape; auto input_shapes = std::get<1>(arg_components); if (input_shapes.has_value()) { TF_RETURN_IF_ERROR(TensorShapeUtils::MakeShape(*input_shapes, &shape)); } else { TF_RETURN_IF_ERROR( TensorShapeUtils::MakeShape(static_cast<int*>(nullptr), 0, &shape)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.h
// Please see the implementation file for documentation of these options. // Import options. extern llvm::cl::opt<std::string> input_arrays; extern llvm::cl::opt<std::string> input_dtypes; extern llvm::cl::opt<std::string> input_shapes; extern llvm::cl::opt<std::string> output_arrays; extern llvm::cl::opt<std::string> control_output_arrays; extern llvm::cl::opt<std::string> inference_type; extern llvm::cl::opt<std::string> min_values;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 10 20:59:50 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
if (input_shape.size() != perm_values.size()) { return false; } SmallVector<int, 8> old_major_index_ordering; SmallVector<int, 8> new_major_index_ordering; for (int i = 0, end = input_shape.size(); i < end; i++) { if (input_shape[i] != 1) { old_major_index_ordering.push_back(i); } if (input_shape[perm_values[i]] != 1) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
signatures=model.matmul_and_same_scale.get_concrete_function( tensor_spec.TensorSpec( shape=input_shape, dtype=dtypes.float32, name='input_tensor' ) ), ) return model def _create_conv2d_model( self, input_shape: Sequence[int], filter_shape: Sequence[int], saved_model_path: str, bias_fn: Optional[ops.Operation] = None,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
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(); } static TranslateToMLIRRegistration GraphdefToMlirTranslate(
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/lite/transforms/optimize_batch_matmul.cc
bmm_op->getLoc(), permuation_tensor_type, DenseElementsAttr::get(permuation_tensor_type, permute)); auto input_shape = input_type.getShape(); llvm::SmallVector<int64_t, 4> permuted_shape(input_shape.begin(), input_shape.end()); // Swaps z dimension and x dimension to get permuted shape. std::iter_swap(permuted_shape.begin() + input_rank - 1,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
// if you want shape inference to consider the input tensors of the // op for shape inference. // - The types need not be set in `input_shapes` as it is not used. // - The number of `input_tensors` should be the same as the number of items // in `input_shapes`. // // The results are returned in `output_shapes` and // `output_resource_shapes_and_types`. The caller is responsible for freeing the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/cc/gradients/math_grad.cc
// [[g1, g1, g1], // [g2, g2, g2]] // input_shape = [2, 3] auto input_shape = Shape(scope, op.input(0)); // output_shape_kept_dims = [2, 1] auto output_shape_kept_dims = ReducedShapeHelper(scope, input_shape, op.input(1)); // This step "flips" any 1s with values from the input_shape, and // replaces remaining entries with 1. This creates a shape that
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 50.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// through entry by entry. ArrayRef<int64_t> input_shape = input_type.getShape(); int input_shape_size = input_shape.size(); Shape slice_sizes(input_shape.begin(), input_shape.end()); int slice_dimensions = slice_sizes.size(); slice_sizes[slice_dimensions - 2] = std::min((int64_t)1, input_shape[input_shape_size - 2]); slice_sizes[slice_dimensions - 1] =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)