- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for set_element_type (0.15 sec)
-
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
TensorShape(), DT_FLOAT, false, tensorflow::XlaLayoutPreference::kNoPreference)); xla::ShapeProto shape_proto; shape_proto.set_element_type(xla::PrimitiveType::F32); shape_proto.mutable_layout(); EXPECT_EQ(shape, xla::Shape(shape_proto)); EXPECT_EQ(options.shape_determination_fns.layout_preference_fn(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.cc
int64_t chunk_size = std::accumulate(std::next(shape.begin(), quantization_dim_ + 1), shape.end(), 1, std::multiplies<int64_t>()); Type new_element_type = IntegerType::get(attr.getContext(), storage_bit_width_); return attr.mapValues(new_element_type, [&](const APFloat &old) { int chunk_index = flatten_index / chunk_size; flatten_index++;
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/tfr/ir/tfr_ops.td
let hasCanonicalizer = 1; } def TFR_GetElementTypeOp : TFR_Op<"get_element_type", [Pure]> { let description = [{ The `get_element_type` operation gets the element type of a tfr.tensor and returns !tfr.attr. Example: ```mlir %1 = "tfr.get_element_type"(%0) : !tfr.tensor -> !tfr.attr %1 = tfr.get_element_type %0 -> !tfr.attr ``` }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/ops.mlir
%2 = shape.to_extent_tensor %1 : !shape.shape -> tensor<2xindex> func.return %2 : tensor<2xindex> } // ----- func.func @get_element_type(%arg0: !tfr.tensor) -> (!tfr.attr, !tfr.attr) { %0 = tfr.get_element_type %arg0 -> !tfr.attr %1 = "tfr.get_element_type"(%arg0) : (!tfr.tensor) -> !tfr.attr func.return %0, %1 : !tfr.attr, !tfr.attr } // ----- // CHECK-LABEL: from_tf_tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 14 22:15:06 UTC 2023 - 13.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
sum_element_type); else result_type = UnrankedTensorType::get(sum_element_type); // Convert if we need enlarge the element type's bitwidth. if (input_element_type != sum_element_type) input_value = rewriter.create<ConvertOp>(op.getLoc(), input_value, sum_element_type); // Create the ReduceWindow op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
[&](RankedTensorType new_element_type) -> bool { DCOMMENT("\t\tConsidering " << new_element_type << " with old " << *potential_element_type); if (!new_element_type || !new_element_type.hasStaticShape()) return false; if (!*potential_element_type) { DCOMMENT("\t\tUpdating potential_element_type " << new_element_type); *potential_element_type = new_element_type;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)