- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 593 for shake (0.04 sec)
-
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
} float CalculateBlockSparsity(const ElementsAttr& attr, const ShapedType& type, const std::vector<int>& block_size) { float sparsity = 0; std::vector<int> shape(2); shape[0] = type.getDimSize(0); shape[1] = type.getDimSize(1); std::vector<int> traversal_order = {}; std::vector<TfLiteDimensionType> format = {}; std::vector<int> b_size = {}; std::vector<int> b_map = {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
// certain downstream optimization passes (e.g. XLA). inline constexpr std::array<int64_t, 4> kOihwToHwioPermutation = {2, 3, 1, 0}; // Returns true if the value has static shape. bool HasStaticShape(Value value); // Returns true if the value has static shape at given dims. bool HasStaticShapeAtDims(Value value, ArrayRef<int> dims); // Whether `value` has known rank of `rank`. Returns false when it is not a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/aot/compile.cc
return errors::Unknown("Couldn't get XLA program shape: ", pshape_or.status().message()); } compile_result->program_shape = pshape_or.value()->ToProto(); xla::ProgramShapeProto* pshape = &compile_result->program_shape; // AotXlaComputationInstance::argument_layouts is a vector of Shape // pointers. Accumulate the Shape objects themselves in a separate vector
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 08:28:57 UTC 2024 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
std::unique_ptr<Graph> g(new Graph(&fld)); tensorflow::TensorProto tensor_proto; tensor_proto.set_dtype(tensorflow::DT_INT32); tensorflow::TensorShapeProto shape; shape.add_dim()->set_size(2); *tensor_proto.mutable_tensor_shape() = shape; for (int i = 0; i < 2; ++i) { tensor_proto.add_int_val(1); } NodeDef const_def; TF_CHECK_OK(NodeDefBuilder("const", "Const")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/fake_quant_per_channel_4bit.pbtxt
node { name: "input" op: "Placeholder" attr { key: "dtype" value { type: DT_FLOAT } } attr { key: "shape" value { shape { dim { size: 1 } dim { size: 1 } dim { size: 1 } dim { size: 256 }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 18.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/canonicalize.mlir
func.func @get_real_shape(%arg0: tensor<1x2xf32>) -> tensor<2xindex> { %0 = "tfr.cast"(%arg0) : (tensor<1x2xf32>) -> !tfr.tensor %1 = tfr.get_shape %0 -> !shape.shape %2 = shape.to_extent_tensor %1 : !shape.shape -> tensor<2xindex> func.return %2 : tensor<2xindex> // CHECK-NEXT: %[[e:.*]] = shape.const_shape [1, 2] : tensor<2xindex> // CHECK-NEXT: return %[[e]] : tensor<2xindex> } // CHECK-LABEL: equal
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
// ReluGrad(gradients, features) = gradients * (features > 0) // The condition that $gradients and $features need to have the same shape is // implicitly enforced: $zero is created to have the same shape as $features, // MHLO_SelectOp enforces that $gradients and $zero have the same shape. def : Pat<(TF_ReluGradOp AnyTensor:$gradients, AnyTensor:$features), (MHLO_SelectOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
for (const std::optional<llvm::ArrayRef<int64_t>>& shape : shapes) { TensorShapeProto& tshape = *shape_list.add_shape(); if (shape.has_value()) { for (int64_t dim : *shape) { tshape.add_dim()->set_size(mlir::ShapedType::isDynamic(dim) ? -1 : dim); } } else { tshape.set_unknown_rank(true); } } // If shape is already set, override it. This can happen if we import
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/cost_analysis.mlir
// expected-remark@+1 {{Cost: 19}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
Value value, ArrayRef<int64_t> shape, Type element_type, Location loc, PatternRewriter& rewriter) { int64_t shape_rank = shape.size(); auto shape_spec_type = RankedTensorType::get({shape_rank}, rewriter.getIntegerType(64)); Type resultType = RankedTensorType::get(shape, element_type); auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); auto shape_tensor =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0)