Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ConvertToTensor (0.26 sec)

  1. 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)
  2. 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)
Back to top