- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for getShape (0.22 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// only when height = width. if (params_type.getShape().size() != 4 || indices_type.getShape().size() != 2) return failure(); if (params_type.getShape()[1] != 1) return failure(); if (params_type.getShape()[0] != params_type.getShape()[2]) return failure(); if (result_type.getShape()[0] != params_type.getShape()[0] * 2) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
} for (auto idx = 0; idx < output_type.getShape().size(); idx++) { if (input_type.getShape()[idx] != output_type.getShape()[idx]) { op.emitOpError() << "the " << idx << "th dim of output tensor is " << output_type.getShape()[idx] << ". It is not equal to the one in input tensor, which is " << input_type.getShape()[idx]; return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
if (input_ty_i.hasStaticShape() && input_ty_i.getShape() != input_ty_0.getShape()) { return op.emitOpError() << "inputs[" << i << "] has shape [" << input_ty_i.getShape() << "] different than the shape of inputs[0]: " << input_ty_0.getShape(); } } if (op.getDimensionsToReduce().size() > input_ty_0.getRank()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
new_values.reserve(num_elements); const auto result_shape = type.getShape(); std::vector<int64_t> current_index(type.getRank(), 0); // Create the new shape with ones padded to the left. const std::vector<int64_t> lhs_new_shape = GetPaddedShape(lhs.getType().getShape(), type.getRank()); const std::vector<int64_t> rhs_new_shape = GetPaddedShape(rhs.getType().getShape(), type.getRank());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Reshape input image to add a new spatial dimension. auto image_type = mlir::cast<ShapedType>(conv_op.getLhs().getType()); SmallVector<int64_t, 4> image_2d_shape(image_type.getShape().begin(), image_type.getShape().end()); image_2d_shape.push_back(1); auto image_2d_type = RankedTensorType::get(image_2d_shape, image_type.getElementType()); auto loc = conv_op.getLoc();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
if (rhs_shape_type.hasRank()) { shape.append(rhs_shape_type.getShape().begin(), rhs_shape_type.getShape().end()); refined_shape = true; } } else if (rhs_shape_type.hasRank()) { for (auto shape_elts : llvm::enumerate( llvm::zip(lhs_shape_type.getShape(), rhs_shape_type.getShape()))) { if (ShapedType::isDynamic(std::get<0>(shape_elts.value())) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
const auto [input_height, input_width] = GetDimSize(op->getOperand(0).getType().cast<ShapedType>().getShape(), dimension_numbers.getInputSpatialDimensions()); const auto [output_height, output_width] = GetDimSize(op->getResult(0).getType().cast<ShapedType>().getShape(), dimension_numbers.getOutputSpatialDimensions());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
".getShape()[" # dim # " ] == " # size>]>; // Returns true if the n-th operand is ranked and has a dimension length <= // size at the rank dim. class TFL_OperandDimIsAtMost<int n, int dim, int size> : And<[ TFL_OperandIsRankedAndHasDimPred<n, dim>, CPred<"$_op.getOperand(" # n # ").getType().cast<ShapedType>()" ".getShape()[" # dim # " ] <= " # size>]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
llvm::ArrayRef<int64_t> shape_ref = mlir::cast<TensorType>(tensor_attr.getType()).getShape(); if (mlir::failed(check_shape(shape_ref))) return std::nullopt; shape = std::vector<int32_t>(shape_ref.begin(), shape_ref.end()); } else if (type.hasRank()) { llvm::ArrayRef<int64_t> shape_ref = type.getShape(); if (mlir::failed(check_shape(shape_ref))) return std::nullopt;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
pkg/kubelet/kubelet_pods_test.go
}, }, }, false, v1.PodRunning, "backoff crashloop container with restart always", }, } for _, test := range tests { status := getPhase(test.pod, test.pod.Status.ContainerStatuses, test.podIsTerminal) assert.Equal(t, test.status, status, "[test %s]", test.test) } } func TestPodPhaseWithRestartAlwaysInitContainers(t *testing.T) {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri Jun 07 00:05:23 UTC 2024 - 198.8K bytes - Viewed (0)