- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 213 for se_shape (0.13 sec)
-
tensorflow/compiler/mlir/lite/transforms/unfold_large_splat_constant.cc
tensorflow::GetTypeFromTFTensorShape( {splat_elements_attr.getType().getRank()}, op_builder->getI64Type()), splat_elements_attr.getType().getShape())); mlir::arith::ConstantOp fill_value = op_builder->create<mlir::arith::ConstantOp>( const_op->getLoc(), DenseElementsAttr::get( tensorflow::GetTypeFromTFTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
TF_RETURN_IF_ERROR( ConvertScalarTypeToDataType(type.getElementType(), &dtype)); if (!type.hasRank()) return std::make_pair(dtype, tensorflow::PartialTensorShape()); auto shape = type.getShape(); llvm::SmallVector<int64_t, 4> dims; dims.assign(shape.begin(), shape.end()); return std::make_pair(dtype, tensorflow::PartialTensorShape(dims)); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
// CHECK: [[VAL_21:%.*]] = "tf.Reshape"([[VAL_15]]#0, [[VAL_20]]) : (tensor<1x10xf32>, tensor<1xi32>) -> tensor<10xf32> // CHECK-DAG: [[VAL_22:%.*]] = "tf.Const"() <{value = dense<-1> : tensor<1xi32>}> : () -> tensor<1xi32> // CHECK: [[VAL_23:%.*]] = "tf.Reshape"([[VAL_15]]#1, [[VAL_22]]) : (tensor<1x10xf32>, tensor<1xi32>) -> tensor<10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
if (!input_type) { return failure(); } const ArrayRef<int64_t> input_shape = input_type.getShape(); const int input_size = input_shape.size(); RankedTensorType begin_type = mlir::cast<RankedTensorType>(begin.getType()); const ArrayRef<int64_t> begin_shape = begin_type.getShape(); const int begin_dim = begin_shape.size(); if (begin_dim != 1) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
return PartialTensorShape(); } if (auto tensor_type = mlir::dyn_cast<mlir::RankedTensorType>(type)) { TensorShapeProto tensor_shape_proto; ConvertToTensorShapeProto(tensor_type.getShape(), &tensor_shape_proto); return PartialTensorShape(tensor_shape_proto); } // If type is not a RankedTensor or UnrankedTensor, it must be a scalar. // Empty TensorShape indicates a scalar.
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/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
// ----- // Tests that a float `stablehlo.reshape` is not converted to `tfl.reshape`. func.func @reshape_float(%arg0: tensor<2x3x4xf32>) -> tensor<6x4xf32> { %0 = stablehlo.reshape %arg0 : (tensor<2x3x4xf32>) -> tensor<6x4xf32> return %0 : tensor<6x4xf32> } // CHECK-LABEL: reshape_float // CHECK-NOT: tfl.reshape // CHECK: stablehlo.reshape // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/validators.cc
} return true; } bool IsDimensionsDegenerateExceptLastOne(TypedAttr val) { if (auto ranked_type = mlir::dyn_cast<RankedTensorType>(val.getType())) { return IsDimensionsDegenerateExceptLastOne(ranked_type.getShape()); } return false; } } // namespace TFL
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/QuantOps.cc
} } // Verify axisStats (optional) attribute. if (getAxisStats()) { if (!getAxis()) return emitOpError("axis must be specified for axisStats"); auto shape = tensorArg.getShape(); auto argSliceSize = std::accumulate(std::next(shape.begin(), *getAxis()), shape.end(), 1, std::multiplies<int64_t>()); auto axisStatsType = getAxisStats()->getShapedType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
auto inputType = mlir::dyn_cast<RankedTensorType>(arg.getType()); // Only rank 1 tensor is supported for now. if (!inputType || inputType.getRank() != 1) continue; auto shape = llvm::to_vector<4>(inputType.getShape()); llvm::SmallVector<int64_t, 4> bounds(shape.begin(), shape.end()); // Mark the dim as dynamic dim. shape[0] = ShapedType::kDynamic; auto extensions =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.cc
} } // Verify axisStats (optional) attribute. if (getAxisStats()) { if (!getAxis()) return emitOpError("axis must be specified for axisStats"); auto shape = tensorArg.getShape(); auto argSliceSize = std::accumulate(std::next(shape.begin(), *getAxis()), shape.end(), 1, std::multiplies<int64_t>()); auto axisStatsType = getAxisStats()->getShapedType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0)