- Sort Score
- Result 10 results
- Languages All
Results 41 - 43 of 43 for shape_of (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
shape.push_back(dim.getSExtValue()); } return tensorflow::GetTypeFromTFTensorShape(shape, etype); } if (auto shape_op = dims.getDefiningOp<ShapeOp>()) { if (auto t = mlir::dyn_cast<ShapedType>(shape_op.getInput().getType())) { return t; } } return UnrankedTensorType::get(etype); }
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/translate/import_model.cc
absl::StatusOr<mlir::Type> ImporterBase::InferOutputType( const Node& node, int idx, mlir::Builder builder) { DataType dtype = node.properties()->output_types[idx]; // Returns output type given inference context. auto shape_ic = [&](shape_inference::InferenceContext* c) -> absl::StatusOr<mlir::Type> { // TODO(b/200093974): Post triage, consider following // GraphConstructor::ValidateShape in checking _output_shapes always.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
const std::vector<int32_t> *shape = nullptr, tflite::TensorType type = tflite::TensorType_FLOAT32, bool has_rank = false) { auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0; return tflite::CreateVariantSubType( _fbb, shape__, type, has_rank); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)