- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 22 for GetSubtypes (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
.create<TF::ReadVariableOp>( loc, ArrayRef<Type>{mlir::cast<TF::ResourceType>( getElementTypeOrSelf(local_var.getType())) .getSubtypes()[0]}, ArrayRef<Value>{local_var}) .getValue(); } // Creates an AssignVariableOp on a local variable. TF::AssignVariableOp WriteLocalVariable(Value local_var, Value value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
auto variant_type = getElementTypeOrSelf(getOperand().getType()).cast<TF::VariantType>(); if (variant_type.getSubtypes().empty()) return {}; return ConvertShapeToAttr(variant_type.getSubtypes()[0], width); } //===----------------------------------------------------------------------===// // TensorListStackOp
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/tensorflow/transforms/tensor_list_ops_decomposition.cc
Type type, llvm::SmallVector<int64_t, 8>* shape) { auto variant_type = mlir::dyn_cast<TF::VariantType>(getElementTypeOrSelf(type)); if (!variant_type || variant_type.getSubtypes().size() != 1) return failure(); TensorType tensor_type = variant_type.getSubtypes().front(); if (!tensor_type.hasStaticShape()) return failure(); for (auto d : tensor_type.getShape()) shape->push_back(d); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
return size.emitOpError("unknown tensor array"); } auto buffer_type = getElementTypeOrSelf(local_var.getType()) .cast<TF::ResourceType>() .getSubtypes()[0] .cast<RankedTensorType>(); OpBuilder builder(size); auto result = cutil::CreateScalarConst(buffer_type.getDimSize(0), builder, size.getLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// Get the data type of the resource. auto subtypes = mlir::cast<TF::ResourceType>( getElementTypeOrSelf(execute.getOperand(it->second))) .getSubtypes(); if (subtypes.size() != 1) continue; auto data_type = getElementTypeOrSelf(subtypes[0]); // The XLA backend does not yet support formatting 64-bit data types.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
if (mlir::isa<mlir::TF::StringType>(type)) return true; auto sub_type = mlir::dyn_cast<mlir::TF::TensorFlowTypeWithSubtype>(type); if (!sub_type) return false; bool has_string = llvm::any_of(sub_type.GetSubtypes(), [](TensorType type) { return mlir::isa<mlir::TF::StringType>(type.getElementType()); }); return has_string; } bool HasStringOperand(Operation& op) { for (auto operand : op.getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (auto resource_type = mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) { llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes(); if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr; for (mlir::TensorType subtype : subtypes) { DataType dtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (auto resource_type = mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) { llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes(); if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr; for (mlir::TensorType subtype : subtypes) { DataType dtype;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
"return getElementTypeOrSelf(element_type());">; let extraClassDeclaration = [{ // Returns type of the TensorList element produced by this op. TensorType element_type() { return handle_dtype().getSubtypes()[0]; } // Returns data type of the result handle. Returned type contains type of // the TensorList element as a subtype. VariantType handle_dtype() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
} // We only support up to one nested type in tf_type.variant_type. if (variant_type.getSubtypes().size() > 1) { return std::nullopt; } if (variant_type.getSubtypes().empty()) { return variant_params; } mlir::TensorType tensor_type = variant_type.getSubtypes().front(); tflite::TensorType tflite_element_type = GetTFLiteType(tensor_type.getElementType()).value();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)