- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ConvertToTensor (0.26 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// shape. The variant type should hold a TensorList. auto proto_attr = op.getValue().dyn_cast<TF::TensorProtoAttr>(); if (!proto_attr) return failure(); tensorflow::Tensor tensor; if (!tensorflow::ConvertToTensor(proto_attr, &tensor).ok()) return failure(); if (tensor.dtype() != tensorflow::DT_VARIANT) return failure(); if (!tensorflow::TensorShapeUtils::IsScalar(tensor.shape())) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
return empty_buffer_; } const_attribute_to_buffer_map_[attr] = index; } // TF doesn't currently support 4-bit types (DT_INT4), so we'll run into // trouble calling ConvertToTensor(). For now, extract the tensor data from // ElementsAttr directly in this and read type from tflite::TensorType instead // of tensorflow::DataType. auto type = mlir::cast<TensorType>(value.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)